Ready for Zero

run_gradio.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import numpy as np
 from PIL import Image, ImageDraw
@@ -341,6 +342,7 @@ def undo_last_point(ref, base):
 
 
 # Main function to run the image processing
+@spaces.GPU
 def run_local(ref, base, *args, progress=gr.Progress(track_tqdm=True)):
     image = Image.fromarray(base)
     ref_image = Image.fromarray(ref)