Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ def unzip_images(zip_file):
|
|
27 |
|
28 |
return image_paths, image_data, session_dir
|
29 |
|
30 |
-
@spaces.GPU(duration=
|
31 |
def generate_caption(image_path, prompt):
|
32 |
try:
|
33 |
# Load processor and model in FP16
|
|
|
27 |
|
28 |
return image_paths, image_data, session_dir
|
29 |
|
30 |
+
@spaces.GPU(duration=120) # Keep increased timeout
|
31 |
def generate_caption(image_path, prompt):
|
32 |
try:
|
33 |
# Load processor and model in FP16
|