Update app.py

app.py

@@ -1,6 +1,10 @@
 import torch
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+# From here down is all the StreamLit UI.
+st.set_page_config(page_title="LangChain Demo", page_icon=":robot:")  # Call it first
+st.header("Hey, I'm your Chat GPT")
+
 # Define SystemMessage class if needed
 class SystemMessage:
     def __init__(self, content):
@@ -17,9 +21,6 @@ model_name = "t5-small"  # Replace with your choice (e.g., facebook/bart-base or
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
-# From here down is all the StreamLit UI.
-st.set_page_config(page_title="LangChain Demo", page_icon=":robot:")
-st.header("Hey, I'm your Chat GPT")
 
 
 if "sessionMessages" not in st.session_state: