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,
|