beriond commited on
Commit
150ee01
路
verified 路
1 Parent(s): 4e008ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from langchain.chains import ConversationalRetrievalChain
9
  from langchain.memory import ConversationBufferMemory
10
 
11
  # 馃敼 Inicializa pipeline com modelo leve
12
- pipe = pipeline("text-generation", model="Qwen/Qwen2.5-1.8B-Instruct", max_new_tokens=256)
13
  llm = HuggingFacePipeline(pipeline=pipe)
14
 
15
  # 馃敼 Mem贸ria de conversa莽茫o
 
9
  from langchain.memory import ConversationBufferMemory
10
 
11
  # 馃敼 Inicializa pipeline com modelo leve
12
+ pipe = pipeline("text2text-generation", model="google/flan-t5-small", max_new_tokens=256
13
  llm = HuggingFacePipeline(pipeline=pipe)
14
 
15
  # 馃敼 Mem贸ria de conversa莽茫o