0

app.py

@@ -8,6 +8,7 @@ from PIL.PngImagePlugin import PngInfo
 import json
 import gradio as gr
 import tempfile
+import spaces
 
 # Set environment variable to reduce memory fragmentation
 os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
@@ -48,6 +49,7 @@ except Exception as e:
     pipe = None
 
 
+@spaces.GPU
 def img2img(
     uploaded_image,
     image_url: str,

requirements.txt

@@ -5,3 +5,4 @@ accelerate>=0.20.0
 gradio>=4.0.0
 Pillow>=9.0.0
 xformers>=0.0.20
+spaces