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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -10
app.py CHANGED
@@ -32,20 +32,11 @@ def load_keybert_model():
32
 
33
  # ─── BlenderBot Response Pipeline ───────────────────────────────────────────
34
  @st.cache_resource
35
- def load_response_pipeline():
36
  # High-level helper using BlenderBot 400M Distill
37
  return pipeline(
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,
46
- tokenizer=seq_tok,
47
- max_new_tokens=150,
48
- do_sample=False
49
  )
50
 
51
  LABEL_MAP = {
 
32
 
33
  # ─── BlenderBot Response Pipeline ───────────────────────────────────────────
34
  @st.cache_resource
35
+ funcdef load_response_pipeline():
36
  # High-level helper using BlenderBot 400M Distill
37
  return pipeline(
38
  "text2text-generation",
39
  model="facebook/blenderbot-400M-distill"
 
 
 
 
 
 
 
 
 
40
  )
41
 
42
  LABEL_MAP = {