beriond commited on
Commit
ad290f4
路
verified 路
1 Parent(s): 9335242

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -26,11 +26,10 @@ def carregar_documento_pdf(path):
26
  # 鈿欙笍 Pipeline com Mistral-7B (贸timo para chat)
27
  pipe = pipeline(
28
  "text-generation",
29
- model="mistralai/Mistral-7B-v0.1",
30
  max_new_tokens=512,
31
- do_sample=True,
32
  temperature=0.7,
33
- repetition_penalty=1.1
34
  )
35
  llm = HuggingFacePipeline(pipeline=pipe)
36
 
 
26
  # 鈿欙笍 Pipeline com Mistral-7B (贸timo para chat)
27
  pipe = pipeline(
28
  "text-generation",
29
+ model="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
30
  max_new_tokens=512,
 
31
  temperature=0.7,
32
+ do_sample=True
33
  )
34
  llm = HuggingFacePipeline(pipeline=pipe)
35