Update app.py

app.py

@@ -25,7 +25,6 @@ with st.sidebar:
     # Dropdown to select model
     model_options = [
         "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
-
     ]
     selected_model = st.selectbox("Select Model", model_options, index=0)
 
@@ -78,7 +77,10 @@ if prompt := st.chat_input("Type your message..."):
         with st.spinner("Generating response..."):
             # Prepare the payload for the API
             payload = {
-                "inputs": prompt,
+                "inputs": {
+                    "system_message": system_message,
+                    "user_message": prompt
+                },
                 "parameters": {
                     "max_new_tokens": max_tokens,
                     "temperature": temperature,
@@ -90,7 +92,7 @@ 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}"
             logger.info(f"Selected model: {selected_model}, API URL: {api_url}")
-
+            print("payload:",payload)
             # Query the Hugging Face API using the selected model
             output = query(payload, api_url)