Update app.py

app.py

@@ -14,36 +14,20 @@ def reset_conversation():
     st.session_state.messages = []
     return None
 
-# Define model links for Hugging Face models
+# Define model links and info
 model_links = {
-    "Mistral": "mistralai/Mistral-7B-Instruct-v0.2",
-    "Gemma-7B": "google/gemma-7b-it",
-    "Gemma-2B": "google/gemma-2b-it",
-    "Zephyr-7B-β": "HuggingFaceH4/zephyr-7b-beta",
-    "Nous-Hermes-2-Yi-34B": "NousResearch/Nous-Hermes-2-Yi-34B"
+    "Addiction recovery AI": "NousResearch/Nous-Hermes-2-Yi-34B",
+    "Mental health AI": "NousResearch/Nous-Hermes-2-Yi-34B"
 }
 
-# Define model info for all models
 model_info = {
-    "Mistral": {
-        'description': "The Mistral model is a Large Language Model (LLM) developed by Mistral AI.",
-        'logo': 'https://mistral.ai/images/logo_hubc88c4ece131b91c7cb753f40e9e1cc5_2589_256x0_resize_q97_h2_lanczos_3.webp'
+    "Addiction recovery AI": {
+        'description': "This model provides support and guidance for individuals on their addiction recovery journey.",
+        'logo': 'https://example.com/addiction_recovery_logo.png'
     },
-    "Gemma-7B": {
-        'description': "The Gemma-7B model is a Large Language Model (LLM) developed by Google with 7 billion parameters.",
-        'logo': 'https://pbs.twimg.com/media/GG3sJg7X0AEaNIq.jpg'
-    },
-    "Gemma-2B": {
-        'description': "The Gemma-2B model is a Large Language Model (LLM) developed by Google with 2 billion parameters.",
-        'logo': 'https://pbs.twimg.com/media/GG3sJg7X0AEaNIq.jpg'
-    },
-    "Zephyr-7B-β": {
-        'description': "The Zephyr-7B-β model is a Large Language Model (LLM) developed by HuggingFace.",
-        'logo': 'https://huggingface.co/HuggingFaceH4/zephyr-7b-alpha/resolve/main/thumbnail.png'
-    },
-    "Nous-Hermes-2-Yi-34B": {
-        'description': "The Nous Hermes model is a Large Language Model (LLM) developed by Nous Research with 34 billion parameters.",
-        'logo': 'https://example.com/nous_hermes_logo.png'
+    "Mental health AI": {
+        'description': "This model offers assistance and resources for individuals dealing with mental health concerns.",
+        'logo': 'https://example.com/mental_health_logo.png'
     }
 }
 
@@ -93,9 +77,8 @@ def interact_with_together_api(messages):
 
     return assistant_response
 
-# Create sidebar with model selection dropdown and temperature slider
+# Create sidebar with model selection dropdown
 selected_model = st.sidebar.selectbox("Select Model", list(model_links.keys()))
-temperature = st.sidebar.slider('Select Temperature', 0.0, 1.0, 0.5)
 st.sidebar.button('Reset Chat', on_click=reset_conversation)
 
 # Display model description and logo