Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import streamlit as st
 import requests
 import logging
@@ -21,15 +20,10 @@ if "messages" not in st.session_state:
 # Sidebar configuration
 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-Distill-Qwen-14B",
-     #   "deepseek-ai/DeepSeek-R1-Distill-Qwen-7B",
-     #   "deepseek-ai/DeepSeek-R1-Distill-Llama-70B", 
-     #   "deepseek-ai/DeepSeek-R1-Distill-Llama-8B"
     ]
     selected_model = st.selectbox("Select Model", model_options, index=0)
 
@@ -82,7 +76,7 @@ if prompt := st.chat_input("Type your message..."):
         with st.spinner("Generating response..."):
             # Prepare the payload for the API
             payload = {
-                "inputs": prompt,
+                "inputs": f"{system_message}\n{prompt}",
                 "parameters": {
                     "max_new_tokens": max_tokens,
                     "temperature": temperature,
@@ -114,4 +108,3 @@ if prompt := st.chat_input("Type your message..."):
     except Exception as e:
         logger.error(f"Application Error: {str(e)}", exc_info=True)
         st.error(f"Application Error: {str(e)}")
-