Update app.py

app.py

@@ -2,15 +2,12 @@ import requests
 import os
 import gradio as gr
 
-# Fetch the API token from the environment variable set in Hugging Face Space Secrets
 API_TOKEN = os.getenv("HF_API_TOKEN")
 if not API_TOKEN:
     raise ValueError("HF_API_TOKEN not found. Please set it in the Space Secrets.")
 
-# DeepSeek-R1 model endpoint
 MODEL_URL = "https://api-inference.huggingface.co/models/deepseek-ai/DeepSeek-R1-Distill-Qwen-32B"
 
-# Headers for authentication
 HEADERS = {
     "Authorization": f"Bearer {API_TOKEN}",
     "Content-Type": "application/json"
@@ -21,15 +18,15 @@ def query_deepseek(prompt):
     payload = {
         "inputs": prompt,
         "parameters": {
-            "max_new_tokens": 100,  # Limit response length
-            "temperature": 0.6,     # Control creativity
-            "top_p": 0.9            # Nucleus sampling
+            "max_new_tokens": 100, 
+            "temperature": 0.6,    
+            "top_p": 0.9           
         }
     }
     
     try:
         response = requests.post(MODEL_URL, headers=HEADERS, json=payload)
-        response.raise_for_status()  # Raise an error for bad status codes
+        response.raise_for_status() 
         result = response.json()
         return result[0]["generated_text"].strip()
     except requests.exceptions.RequestException as e:
@@ -43,7 +40,6 @@ def chat_interface(user_input, history):
     response = query_deepseek(prompt)
     return response
 
-# Custom CSS for blue buttons
 css = """
 button {
     background-color: #007bff !important;  /* Blue background */
@@ -57,7 +53,7 @@ button:hover {
 }
 """
 
-# Create Gradio interface (no unsupported arguments)
+
 interface = gr.ChatInterface(
     fn=chat_interface,
     title="LocoBot : Your AI Companion",