beriond commited on
Commit
306b9fc
路
verified 路
1 Parent(s): 3b87493

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
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
- "text2text-generation",
15
- model="mistralai/Mistral-7B-Instruct",
16
- max_new_tokens=1024,
17
  do_sample=True,
18
- temperature=0.3,
19
- repetition_penalty=1.1
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