Update app.py

app.py

@@ -1,12 +1,25 @@
 import gradio as gr
+import os
+
+# Optional: set your HF token in the environment (for private Spaces)
+HF_TOKEN = os.getenv("HF_TOKEN")  # set this in Space Secrets or locally
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Zephyr-7B-Beta Public API")
-    
-    # Load hosted model without login
-    gr.load(
-        "models/HuggingFaceH4/zephyr-7b-beta",
-        provider="auto"  # no accept_token, no login
-    )
-
-demo.launch()
+    gr.Markdown("# Zephyr-7B-Beta Chat API")
+
+    # Load the hosted model
+    if HF_TOKEN:
+        # If token exists, use it
+        gr.load(
+            "models/HuggingFaceH4/zephyr-7b-beta",
+            provider="auto",
+            hf_token=HF_TOKEN
+        )
+    else:
+        # Public access
+        gr.load(
+            "models/HuggingFaceH4/zephyr-7b-beta",
+            provider="auto"
+        )
+
+demo.launch(server_name="0.0.0.0", server_port=7860)