Update app.py

app.py

@@ -20,6 +20,7 @@ import os
 
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
+st.set_page_config(page_title="TeapotAI Chat", page_icon=":robot_face:", layout="wide")
 tokenizer = None
 model = None
 model_name = "teapotai/teapotllm"
@@ -27,8 +28,6 @@ with st.spinner('Loading Model...'):
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
-    
-
 
 
 def log_time(func):
@@ -114,7 +113,6 @@ def handle_chat(user_prompt, user_input, teapot_ai):
 
 
 def main():
-    st.set_page_config(page_title="TeapotAI Chat", page_icon=":robot_face:", layout="wide")
     
     st.sidebar.header("Retrieval Augmented Generation")
     user_prompt = st.sidebar.text_area("Enter prompt, leave empty for search")