Update app.py

app.py

@@ -1,8 +1,13 @@
 import streamlit as st
 from transformers import pipeline
 
-# Initialize the model
-pipe = pipeline("text2text-generation", model="facebook/blenderbot-400M-distill")
+# Cache the model pipeline to avoid reloading it on each rerun
+@st.cache_resource
+def load_model():
+    return pipeline("text2text-generation", model="facebook/blenderbot-400M-distill")
+
+# Load the model (cached)
+pipe = load_model()
 conversation_history = ""
 
 # Define the conversation function