captain-awesome
commited on
Commit
•
0abcd0d
1
Parent(s):
2da7358
Update app.py
Browse files
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 |
-
|
142 |
-
|
143 |
-
|
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",
|