Update demo/gradio_demo.py

demo/gradio_demo.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import spaces
 from PIL import Image, ImageDraw, ImageFont
 import re
 import numpy as np
@@ -177,7 +178,7 @@ def extract_bbox_and_original_image(edited_image):
         print(f"Unknown input type: {type(edited_image)}")
         return None, []
 
-
+@spaces.GPU
 def process(image, example_image, prompt, threshold):
     image, bbox_list = extract_bbox_and_original_image(image)