Update app.py

app.py

@@ -1,13 +1,13 @@
 import streamlit as st
 import requests
 
-# Hugging Face API URL
+# 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):
+def query(payload, api_url):
     headers = {"Authorization": f"Bearer {st.secrets['HF_TOKEN']}"}
-    response = requests.post(API_URL, headers=headers, json=payload)
+    response = requests.post(api_url, headers=headers, json=payload)
     return response.json()
 
 # Page configuration
@@ -26,6 +26,14 @@ with st.sidebar:
     st.header("Model Configuration")
     st.markdown("[Get HuggingFace Token](https://huggingface.co/settings/tokens)")
 
+    # Dropdown to select model
+    model_options = [
+        "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
+        "deepseek-ai/DeepSeek-R1",
+        "deepseek-ai/DeepSeek-R1-Zero"
+    ]
+    selected_model = st.selectbox("Select Model", model_options, index=0)
+
     system_message = st.text_area(
         "System Message",
         value="You are a friendly Chatbot created by ruslanmv.com",
@@ -76,15 +84,15 @@ if prompt := st.chat_input("Type your message..."):
                 }
             }
 
-            # Query the Hugging Face API
-            output = query(payload)
+            # Query the Hugging Face API using the selected model
+            output = query(payload, f"https://api-inference.huggingface.co/models/{selected_model}")
 
             # Handle API response
             if isinstance(output, list) and len(output) > 0 and 'generated_text' in output[0]:
                 assistant_response = output[0]['generated_text']
             else:
                 st.error("Error: Unable to generate a response. Please try again.")
-                return  # Return from the function to stop further execution
+                continue  # Skip further execution for this iteration
 
             with st.chat_message("assistant"):
                 st.markdown(assistant_response)