mayf commited on
Commit
a6c9351
·
verified ·
1 Parent(s): 601d686

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -6
app.py CHANGED
@@ -35,15 +35,10 @@ from transformers import Conversation
35
 
36
  @st.cache_resource
37
  def load_response_pipeline():
38
- # Use conversational pipeline for interactive-style replies
39
  return pipeline(
40
  "conversational",
41
  model="facebook/blenderbot-400M-distill"
42
- ):
43
- # High-level helper using BlenderBot 400M Distill
44
- return pipeline(
45
- "text2text-generation",
46
- model="facebook/blenderbot-400M-distill"
47
  )
48
 
49
  LABEL_MAP = {
 
35
 
36
  @st.cache_resource
37
  def load_response_pipeline():
38
+ # High-level conversational helper using BlenderBot 400M Distill
39
  return pipeline(
40
  "conversational",
41
  model="facebook/blenderbot-400M-distill"
 
 
 
 
 
42
  )
43
 
44
  LABEL_MAP = {