Update app.py

app.py

@@ -18,13 +18,14 @@ else:
     default_port = 8861  # Local default port
 
 
-try:
-    import spaces
-    GPU = spaces.GPU
-except ImportError:
-    # Define a no-op decorator if running locally
-    def GPU(func):
-        return func
+# try:
+#     import spaces
+#     GPU = spaces.GPU
+# except ImportError:
+#     # Define a no-op decorator if running locally
+#     def GPU(func):
+#         return func
+import spaces
 
 # Create model directories if they don't exist
 os.makedirs("models", exist_ok=True)
@@ -33,7 +34,7 @@ os.makedirs("stimuli", exist_ok=True)
 # Initialize model
 model = GenerativeInferenceModel()
 
-@GPU 
+@spaces.GPU 
 def run_inference(image, model_type, illusion_type, eps_value, num_iterations):
     # Convert eps to float
     eps = float(eps_value)