Spaces:
Running
on
Zero
Running
on
Zero
Update gradio_ui.py
Browse files- gradio_ui.py +1 -1
gradio_ui.py
CHANGED
@@ -178,7 +178,7 @@ def blip_image_captioning(image: PIL.Image.Image,
|
|
178 |
# return caption
|
179 |
|
180 |
# Define a function to process the image with the loaded model
|
181 |
-
@spaces.GPU
|
182 |
def process_image(image_path: str,
|
183 |
controlnet_model_name_or_path: str,
|
184 |
caption_model_name: str,
|
|
|
178 |
# return caption
|
179 |
|
180 |
# Define a function to process the image with the loaded model
|
181 |
+
@spaces.GPU(duration=100)
|
182 |
def process_image(image_path: str,
|
183 |
controlnet_model_name_or_path: str,
|
184 |
caption_model_name: str,
|