arampacha commited on
Commit
ca77bef
1 Parent(s): 3d7233c
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -6,7 +6,11 @@ import time
6
 
7
  model_id = "arampacha/DialoGPT-medium-simpsons"
8
 
9
- dialog = pipeline("conversational", model=model_id)
 
 
 
 
10
 
11
  parameters = {
12
  "min_length":None,
 
6
 
7
  model_id = "arampacha/DialoGPT-medium-simpsons"
8
 
9
+ @st.cache(allow_output_mutation=True)
10
+ def get_pipeline():
11
+ return pipeline("conversational", model=model_id)
12
+
13
+ dialog = get_pipeline()
14
 
15
  parameters = {
16
  "min_length":None,