From the beginning :_)

app.py

@@ -1,18 +1,13 @@
 import streamlit as st
 from transformers import pipeline
 
-# Load your LLM
-chatbot = pipeline("text-generation", model="gpt2")
+# Load the model
+chatbot = pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B")
 
-# Streamlit app interface
-st.title("Chat with LLM")
-
-user_input = st.text_input("You: ", placeholder="Type your message here...")
+# Streamlit app
+st.title("Chat with an LLM")
+user_input = st.text_input("Your Message:")
 
 if st.button("Send"):
-    if user_input.strip():
-        with st.spinner("Generating response..."):
-            response = chatbot(user_input, max_length=100, num_return_sequences=1)
-            st.text_area("AI:", value=response[0]["generated_text"], height=200, max_chars=None)
-    else:
-        st.warning("Please enter a valid input.")
+    response = chatbot(user_input, max_length=100, do_sample=True)
+    st.text_area("LLM Response:", response[0]["generated_text"], height=200)