Update app.py

app.py

@@ -9,14 +9,7 @@ MODEL_CHOICES = {
 }
 
 
-def complete_text(
-    prompt: str,
-    max_tokens: int,
-    temperature: float,
-    top_p: float,
-    model_choice: str,
-    hf_token: gr.OAuthToken,  # token provided after user logs in
-):
+def complete_text(prompt, max_tokens, temperature, top_p, model_choice, hf_token):
     """
     Generate a text completion from a Hugging Face model, streamed chunk by chunk.
     """
@@ -24,8 +17,7 @@ def complete_text(
         yield "⚠️ Please enter a prompt."
         return
 
-    # If the user has not logged in, warn them (needed for gated models like Mistral / LLaMA)
-    if not getattr(hf_token, "token", None):
+    if not hf_token or not getattr(hf_token, "token", None):
         yield "🔐 Please login with your Hugging Face account (see left sidebar)."
         return
 
@@ -57,11 +49,7 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         with gr.Column(scale=2):
-            prompt = gr.Textbox(
-                label="Prompt",
-                placeholder="Type the beginning of your text...",
-                lines=4,
-            )
+            prompt = gr.Textbox(label="Prompt", placeholder="Type the beginning of your text...", lines=4)
             max_tokens = gr.Slider(1, 1024, value=100, step=1, label="Max new tokens")
             temperature = gr.Slider(0.0, 2.0, value=0.7, step=0.1, label="Temperature")
             top_p = gr.Slider(0.1, 1.0, value=1.0, step=0.05, label="Top-p")
@@ -73,10 +61,10 @@ with gr.Blocks() as demo:
             submit = gr.Button("Generate Completion")
         with gr.Column(scale=3):
             with gr.Sidebar():
-                login = gr.LoginButton()  # supplies OAuth token at runtime
+                login = gr.LoginButton()  # supplies OAuth token
             output = gr.Textbox(label="Generated Completion", lines=15)
 
-    # Wire inputs (6 args) to match function params (6 args)
+    # Function has 6 args, so inputs list must have exactly 6 items
     submit.click(
         fn=complete_text,
         inputs=[prompt, max_tokens, temperature, top_p, model_choice, login],