Update app.py

app.py

@@ -1,8 +1,11 @@
 import streamlit as st
 from transformers import pipeline
 
-# Load the model pipeline
-@st.cache_resource  # Cache the model to avoid reloading
+# Set Streamlit page configuration
+st.set_page_config(page_title="AI Chatbot", layout="centered")
+
+# Load the model pipeline (cached to avoid reloading on each run)
+@st.cache_resource
 def load_pipeline():
     model_name = "Orenguteng/Llama-3.1-8B-Lexi-Uncensored-V2"
     return pipeline("text-generation", model=model_name)
@@ -10,28 +13,31 @@ def load_pipeline():
 pipe = load_pipeline()
 
 # App UI
-st.set_page_config(page_title="Hugging Face Chatbot", layout="centered")
 st.title("🤖 AI Chatbot")
 st.markdown(
     """
-    Welcome to the **AI Chatbot** powered by Hugging Face's Llama-3.1-8B-Lexi-Uncensored-V2 model. 
-    Enter your message below to get started!
+    Welcome to the **AI Chatbot** powered by Hugging Face's **Llama-3.1-8B-Lexi-Uncensored-V2** model.  
+    Enter your message below, and the AI will respond!
     """
 )
 
-# Input Box
+# Input Textbox
 user_input = st.text_area(
-    "Your Message", 
-    placeholder="Type your message here...", 
+    "Your Message",
+    placeholder="Type your message here...",
     height=100
 )
 
-# Generate Button
+# Generate Button and Response
 if st.button("Generate Response"):
     if user_input.strip():
         with st.spinner("Generating response..."):
+            # Generate a response
             response = pipe(user_input, max_length=150, num_return_sequences=1)
             st.text_area("Response", value=response[0]['generated_text'], height=200)
     else:
         st.warning("Please enter a message before clicking the button.")
-        
+
+# Footer
+st.markdown("---")
+st.markdown("Made with ❤️ using [Streamlit](https://streamlit.io) and [Hugging Face](https://huggingface.co).")