Upload folder using huggingface_hub

app.py

@@ -160,12 +160,22 @@ with gr.Blocks(title="LightDiffusion Web UI") as demo:
         outputs=gallery
     )
 
+def is_huggingface_space():
+    return "SPACE_ID" in os.environ
+
 # For local testing
 if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=8000,
-        auth=None,
-        share=True,
-        debug=True
-    )
+    if is_huggingface_space():
+        demo.launch(
+            debug=False,
+            server_name="0.0.0.0",
+            server_port=7860  # Standard HF Spaces port
+        )
+    else:
+        demo.launch(
+            server_name="0.0.0.0", 
+            server_port=8000,
+            auth=None,
+            share=True,  # Only enable sharing locally
+            debug=True
+        )