Update app.py

app.py

@@ -19,6 +19,7 @@ client = OpenAI(
 #Create supported models
 model_links ={
     "HAH-2024-v0.1":"drmasad/HAH-2024-v0.11",
+    "Mistral":"mistralai/Mistral-7B-Instruct-v0.2",
 }
 
 model_info ={
@@ -26,6 +27,11 @@ model_info ={
         {'description':"""The HAH-2024-v0.1 model is a **Large Language Model (LLM)** that's able to have question and answer interactions.\n \
             \nIt was created by fine tuning Mistra 7b instruct usin 3000 review articles on diabetes** \n""",
         'logo':'https://www.hmgaihub.com/untitled.png'},
+    "Mistral":
+    {'description':
+        """The Mistral model is a **Large Language Model (LLM)** that's able to have question and answer interactions...""",
+    'logo':
+        'https://mistral.ai/images/logo_hubc88c4ece131b91c7cb753f40e9e1cc5_2589_256x0_resize_q97_h2_lanczos_3.webp'},
 }
 
 def reset_conversation():
@@ -88,7 +94,7 @@ for message in st.session_state.messages:
         st.markdown(message["content"])
 
 # Accept user input
-if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question on diabetes"):
+if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
 
     # Display user message in chat message container
     with st.chat_message("user"):
@@ -101,8 +107,11 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question on diabe
     with st.chat_message("assistant"):
         stream = client.chat.completions.create(
             model=model_links[selected_model],
-            messages=[{"role": m["role"], "content": m["content"]} for m in st.session_state.messages],
-            temperature=temp_values,
+            messages=[
+                {"role": m["role"], "content": m["content"]}
+                for m in st.session_state.messages
+            ],
+            temperature=temp_values,#0.5,
             stream=True,
             max_tokens=3000,
         )