Update app.py

app.py

@@ -6,6 +6,7 @@ from huggingface_hub import login
 # Fetch token from environment (automatically loaded from secrets)
 hf_token = os.getenv("gemma3")
 login(hf_token)
+
 # Initialize the client with your model
 client = InferenceClient("hackergeek98/gemma-finetuned")
 
@@ -26,11 +27,11 @@ def respond(
             prompt += f"Assistant: {assistant_msg}\n"
     prompt += f"User: {message}\nAssistant: "
 
-    # Call the text generation API with the correct 'prompt' parameter
+    # Call the text generation API with updated parameter name
     response = client.text_generation(
         model="hackergeek98/gemma-finetuned",
         prompt=prompt,
-        max_tokens=max_tokens,
+        max_new_tokens=max_tokens,
         temperature=temperature,
         top_p=top_p,
     )
@@ -52,3 +53,4 @@ if __name__ == "__main__":
     demo.launch()
 
 
+