Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,13 +11,12 @@ from langchain.prompts import PromptTemplate
|
|
| 11 |
|
| 12 |
# 鈿欙笍 Modelo mais robusto para Spaces com GPU (pode mudar para mistralai/Mistral-7B-Instruct ou zephyr-7b-beta)
|
| 13 |
pipe = pipeline(
|
| 14 |
-
|
| 15 |
-
model="
|
| 16 |
-
max_new_tokens=
|
| 17 |
do_sample=True,
|
| 18 |
-
temperature=0.3
|
| 19 |
-
|
| 20 |
-
)
|
| 21 |
llm = HuggingFacePipeline(pipeline=pipe)
|
| 22 |
|
| 23 |
# 馃 Mem贸ria de curto prazo para manter conversas coerentes
|
|
|
|
| 11 |
|
| 12 |
# 鈿欙笍 Modelo mais robusto para Spaces com GPU (pode mudar para mistralai/Mistral-7B-Instruct ou zephyr-7b-beta)
|
| 13 |
pipe = pipeline(
|
| 14 |
+
"text2text-generation",
|
| 15 |
+
model="google/flan-t5-large",
|
| 16 |
+
max_new_tokens=512,
|
| 17 |
do_sample=True,
|
| 18 |
+
temperature=0.3
|
| 19 |
+
)
|
|
|
|
| 20 |
llm = HuggingFacePipeline(pipeline=pipe)
|
| 21 |
|
| 22 |
# 馃 Mem贸ria de curto prazo para manter conversas coerentes
|