Update app.py

app.py

@@ -6,16 +6,18 @@ import os
 import argparse
 from inference import GenerativeInferenceModel, get_inference_configs
 
+# 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", 8860))  # Use provided PORT or fallback to 7860
+else:
+    default_port = 7860  # Local default port
+
 # Parse command line arguments
 parser = argparse.ArgumentParser(description='Run Generative Inference Demo')
-parser.add_argument('--port', type=int, default=7860, help='Port to run the server on')
+parser.add_argument('--port', type=int, default=default_port, 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: