Update app.py

app.py

@@ -1,15 +1,6 @@
 import streamlit as st
 import requests
 
-# Hugging Face API URL (default model)
-API_URL = "https://api-inference.huggingface.co/models/deepseek-ai/DeepSeek-R1-Distill-Qwen-32B"
-
-# Function to query the Hugging Face API
-def query(payload, api_url):
-    headers = {"Authorization": f"Bearer {st.secrets['HF_TOKEN']}"}
-    response = requests.post(api_url, headers=headers, json=payload)
-    return response.json()
-
 # Page configuration
 st.set_page_config(
     page_title="DeepSeek Chatbot - ruslanmv.com",
@@ -55,6 +46,12 @@ with st.sidebar:
         0.1, 1.0, 0.9
     )
 
+# Function to query the Hugging Face API
+def query(payload, api_url):
+    headers = {"Authorization": f"Bearer {st.secrets['HF_TOKEN']}"}
+    response = requests.post(api_url, headers=headers, json=payload)
+    return response.json()
+
 # Chat interface
 st.title("🤖 DeepSeek Chatbot")
 st.caption("Powered by Hugging Face Inference API - Configure in sidebar")
@@ -84,8 +81,11 @@ if prompt := st.chat_input("Type your message..."):
                 }
             }
 
+            # Dynamically construct the API URL based on the selected model
+            api_url = f"https://api-inference.huggingface.co/models/{selected_model}"
+
             # Query the Hugging Face API using the selected model
-            output = query(payload, f"https://api-inference.huggingface.co/models/{selected_model}")
+            output = query(payload, api_url)
 
             # Handle API response
             if isinstance(output, list) and len(output) > 0 and 'generated_text' in output[0]:
@@ -99,4 +99,4 @@ if prompt := st.chat_input("Type your message..."):
                 st.error("Error: Unable to generate a response. Please try again.")
 
     except Exception as e:
-        st.error(f"Application Error: {str(e)}")
+        st.error(f"Application Error: {str(e)}")