caching

app.py

@@ -6,7 +6,11 @@ import time
 
 model_id = "arampacha/DialoGPT-medium-simpsons"
 
-dialog = pipeline("conversational", model=model_id)
+@st.cache(allow_output_mutation=True)
+def get_pipeline():
+    return pipeline("conversational", model=model_id)
+
+dialog = get_pipeline()
 
 parameters = {
     "min_length":None,