Qdssa commited on
Commit
9374b99
·
verified ·
1 Parent(s): 69f7ee8

Update gradio_ui.py

Browse files
Files changed (1) hide show
  1. 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,