harrim-nv commited on
Commit
4cdf7e3
·
verified ·
1 Parent(s): 59f0fba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -97,6 +97,7 @@ except Exception as e:
97
  print(f"Error loading processor: {e}")
98
  raise
99
 
 
100
  def process_input(image, video, text_prompt, temperature, top_p, repetition_penalty, max_tokens, role, custom_role_text):
101
  """Process the input and generate a response."""
102
  try:
 
97
  print(f"Error loading processor: {e}")
98
  raise
99
 
100
+ @spaces.GPU
101
  def process_input(image, video, text_prompt, temperature, top_p, repetition_penalty, max_tokens, role, custom_role_text):
102
  """Process the input and generate a response."""
103
  try: