mayf commited on
Commit
0bba77d
·
verified ·
1 Parent(s): 5ffa6af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -38,8 +38,8 @@ def load_response_pipeline():
38
  "text2text-generation",
39
  model="facebook/blenderbot-400M-distill"
40
  ):
41
- seq_tok = AutoTokenizer.from_pretrained("google/flan-t5-base")
42
- seq_model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
43
  return pipeline(
44
  "text2text-generation",
45
  model=seq_model,
 
38
  "text2text-generation",
39
  model="facebook/blenderbot-400M-distill"
40
  ):
41
+ seq_tok = AutoTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
42
+ seq_model = AutoModelForSeq2SeqLM.from_pretrained("facebook/blenderbot-400M-distill")
43
  return pipeline(
44
  "text2text-generation",
45
  model=seq_model,