fffiloni commited on
Commit
2702839
1 Parent(s): 33afdce

add Zero GPU capabilities

Browse files
Files changed (1) hide show
  1. gradio_ui.py +2 -0
gradio_ui.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import PIL
2
  import torch
3
  import subprocess
@@ -177,6 +178,7 @@ def blip_image_captioning(image: PIL.Image.Image,
177
  # return caption
178
 
179
  # Define a function to process the image with the loaded model
 
180
  def process_image(image_path: str,
181
  controlnet_model_name_or_path: str,
182
  caption_model_name: str,
 
1
+ import spaces
2
  import PIL
3
  import torch
4
  import subprocess
 
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,