Update app.py

app.py

@@ -1,7 +1,6 @@
 import streamlit as st
 from openai import OpenAI
 import os
-import numpy as np
 from dotenv import load_dotenv
 import random
 
@@ -30,21 +29,21 @@ model_links = {
     "Gemma-7b-it": "google/gemma-7b-it",
 }
 
-# Pull info about the model to display
+# Model information including logos
 model_info = {
     "Meta-Llama-3.1-8B": {
         'description': """The Llama (3.1) model is a **Large Language Model (LLM)** that's able to have question and answer interactions.
-        \nIt was created by the [**Meta's AI**](https://llama.meta.com/) team and has over **8 billion parameters.**\n"""
+        \nIt was created by the [**Meta's AI**](https://llama.meta.com/) team and has over **8 billion parameters.**\n""",
         "logo": "llama_logo.gif",
     },
     "Mistral-7B-Instruct-v0.3": {
         'description': """The Mistral-7B-Instruct-v0.3 is an instruct-tuned version of Mistral-7B.
-        \nIt was created by [**Mistral AI**](https://mistral.ai/) and has **7 billion parameters.**\n"""
+        \nIt was created by [**Mistral AI**](https://mistral.ai/) and has **7 billion parameters.**\n""",
         "logo": "mistrail.jpeg",
     },
     "Gemma-7b-it": {
         'description': """Gemma is a family of lightweight, state-of-the-art open models from Google.
-        \nThe 7B-it variant is instruction-tuned and has **7 billion parameters.**\n"""
+        \nThe 7B-it variant is instruction-tuned and has **7 billion parameters.**\n""",
         "logo": "gemma.jpeg",
     }
 }
@@ -61,9 +60,10 @@ def main():
 
     st.markdown(f'_powered_ by ***:violet[{selected_model}]***')
 
-    # Display model info
+    # Display model info and logo
     st.sidebar.write(f"You're now chatting with **{selected_model}**")
     st.sidebar.markdown(model_info[selected_model]['description'])
+    st.sidebar.image(model_info[selected_model]['logo'], use_column_width=True)
     st.sidebar.markdown("*Generated content may be inaccurate or false.*")
 
     # Initialize chat history