Fix 500 error: add error handling, lazy initialization, and fix Gradio launch config for Hugging Face Spaces

app.py

@@ -34,8 +34,14 @@ else:
 # Use command line port if provided, otherwise use default
 server_port = args.port if args.port is not None else default_port
 
-# Initialize model
-model = GenerativeInferenceModel()
+# Initialize model (lazy initialization to avoid startup failures)
+try:
+    model = GenerativeInferenceModel()
+    print("Model manager initialized successfully")
+except Exception as e:
+    print(f"Warning: Error initializing model manager: {e}")
+    print("Will attempt to initialize on first use")
+    model = None
 
 # Define example images and their parameters with updated values from the research
 examples = [
@@ -216,6 +222,15 @@ examples = [
 @GPU 
 def run_inference(image, model_type, inference_type, eps_value, num_iterations, 
                  initial_noise=0.05, diffusion_noise=0.3, step_size=0.8, model_layer="layer3"):
+    # Initialize model if not already initialized
+    global model
+    if model is None:
+        try:
+            model = GenerativeInferenceModel()
+            print("Model manager initialized on first use")
+        except Exception as e:
+            return None, f"Error initializing model: {str(e)}. Please try again."
+    
     # Check if image is provided
     if image is None:
         return None, "Please upload an image before running inference."
@@ -486,9 +501,13 @@ with gr.Blocks(title="Generative Inference for Psychiatry Demo", css="""
 # Launch the demo
 if __name__ == "__main__":
     print(f"Starting server on port {server_port}")
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=server_port,
-        share=False,
-        debug=True
-    )
+    # On Hugging Face Spaces, don't specify server_name/server_port
+    if "SPACE_ID" in os.environ:
+        demo.launch(share=False, debug=False)
+    else:
+        demo.launch(
+            server_name="0.0.0.0",
+            server_port=server_port,
+            share=False,
+            debug=True
+        )

inference.py

@@ -229,7 +229,7 @@ class GenerativeInferenceModel:
         """Get ImageNet labels."""
         url = "https://raw.githubusercontent.com/anishathalye/imagenet-simple-labels/master/imagenet-simple-labels.json"
         try:
-            response = requests.get(url)
+            response = requests.get(url, timeout=10)  # Add timeout to prevent hanging
             if response.status_code == 200:
                 return response.json()
             else: