Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 from openai import OpenAI
+import torch
 import os
 import sys
 from dotenv import load_dotenv, dotenv_values
@@ -62,7 +63,6 @@ if "prev_option" not in st.session_state:
 
 if st.session_state.prev_option != selected_model:
     st.session_state.messages = []
-    st.write(f"Changed to {selected_model}")
     st.session_state.prev_option = selected_model
 
 # Pull in the model we want to use