Update app.py

app.py

@@ -5,6 +5,7 @@ import PIL.Image as Image
 import numpy as np
 import os
 from huggingface_hub import hf_hub_download
+import spaces
 
 token = os.getenv("ACE_TOKEN")
 repo_id = "LexBwmn/ACE_LAB"
@@ -20,6 +21,7 @@ except Exception as e:
     print(f"Error loading private model: {e}")
     model = None
 
+@spaces.GPU
 def predict(img):
     global model
     if model is None: