sitammeur commited on
Commit
936e343
1 Parent(s): 0e9415c

Update src/app/model.py

Browse files
Files changed (1) hide show
  1. src/app/model.py +1 -1
src/app/model.py CHANGED
@@ -36,7 +36,7 @@ except Exception as e:
36
  raise CustomExceptionHandling(e, sys) from e
37
 
38
 
39
- @spaces.GPU(duration=120)
40
  def run_example(
41
  task_prompt: str, image: Image.Image, text_input: Optional[str] = None
42
  ) -> str:
 
36
  raise CustomExceptionHandling(e, sys) from e
37
 
38
 
39
+ @spaces.GPU
40
  def run_example(
41
  task_prompt: str, image: Image.Image, text_input: Optional[str] = None
42
  ) -> str: