Update app.py

app.py

@@ -2,6 +2,13 @@ import streamlit as st
 import requests
 import datetime
 
+API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.1"
+headers = {"Authorization": "Bearer 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"}
+
+def query(payload):
+	response = requests.post(API_URL, headers=headers, json=payload, verify=False)
+	return response.json()
+
 st.header("#CodeWars localGPT", divider='rainbow')
 
 st.markdown("*#CodeWars* is **really** ***cool***... :sunglasses: our focus is on quality reasoning.")
@@ -16,7 +23,8 @@ if prompt:
     with st.chat_message(option):
          st.write(f"{datetime.datetime.now()} :red[{option}:] ", prompt)
     
-    response = requests.get("https://dummyjson.com/products/1").text
+    response = query({
+        "inputs": prompt,
+    })
 
-    with st.chat_message("Bot"):
-         st.write(f"{datetime.datetime.now()}", response) 
+    st.write(f"{datetime.datetime.now()}", response)