captain-awesome commited on
Commit
0abcd0d
1 Parent(s): 2da7358

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -137,11 +137,12 @@ def get_response(user_input):
137
  # task="text2text-generation",
138
  # # model_kwargs={"temperature": 0.2},
139
  # )
140
- llm = HuggingFacePipeline.from_model_id(
141
- model_id="google-t5/t5-small",
142
- task="text2text-generation",
143
- # model_kwargs={"temperature": 0.2},
144
- )
 
145
 
146
  # llm = HuggingFacePipeline.from_model_id(
147
  # model_id="lmsys/fastchat-t5-3b-v1.0",
 
137
  # task="text2text-generation",
138
  # # model_kwargs={"temperature": 0.2},
139
  # )
140
+ # llm = HuggingFacePipeline.from_model_id(
141
+ # model_id="google-t5/t5-small",
142
+ # task="text2text-generation",
143
+ # # model_kwargs={"temperature": 0.2},
144
+ # )
145
+ llm = pipeline(task="conversational", model="facebook/blenderbot-400M-distill")
146
 
147
  # llm = HuggingFacePipeline.from_model_id(
148
  # model_id="lmsys/fastchat-t5-3b-v1.0",