Update app.py

app.py

@@ -1,5 +1,4 @@
 import gradio as gr
-import spaces
 import torch
 import numpy as np
 from PIL import Image
@@ -12,6 +11,20 @@ parser = argparse.ArgumentParser(description='Run Generative Inference Demo')
 parser.add_argument('--port', type=int, default=7860, help='Port to run the server on')
 args = parser.parse_args()
 
+# Check if running on Hugging Face Spaces (using 'SPACE_ID' as an example environment variable)
+if "SPACE_ID" in os.environ:
+    default_port = int(os.environ.get("PORT", 7860))  # Use provided PORT or fallback to 7860
+else:
+    default_port = 8861  # Local default port
+
+
+try:
+    from spaces import GPU
+except ImportError:
+    # Define a no-op decorator if running locally
+    def GPU(func):
+        return func
+
 # Create model directories if they don't exist
 os.makedirs("models", exist_ok=True)
 os.makedirs("stimuli", exist_ok=True)
@@ -19,7 +32,7 @@ os.makedirs("stimuli", exist_ok=True)
 # Initialize model
 model = GenerativeInferenceModel()
 
-@spaces.GPU 
+@GPU 
 def run_inference(image, model_type, illusion_type, eps_value, num_iterations):
     # Convert eps to float
     eps = float(eps_value)