Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
-MODEL_NAME = "distilgpt2"  # you can change this later
+MODEL_NAME = "Qwen/Qwen2.5-3B-Instruct" 
 
 @st.cache_resource
 def load_pipeline():
@@ -16,10 +16,10 @@ def load_pipeline():
 
 gen_pipeline, tokenizer = load_pipeline()
 
-st.set_page_config(page_title="Simple Chat LLM", page_icon="💬")
+st.set_page_config(page_title="Ask Anything!", page_icon="💬")
 
-st.title("💬 Simple Chat LLM")
-st.caption("Streamlit app running on Hugging Face Spaces")
+st.title("💬 Simple LLM")
+# st.caption("Streamlit app running on Hugging Face Spaces")
 
 # Initialize chat history
 if "messages" not in st.session_state: