CHUNYU0505 commited on
Commit
b867ef7
·
verified ·
1 Parent(s): 0abac0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -55,9 +55,11 @@ llm = HuggingFaceEndpoint(
55
  repo_id="google/flan-t5-large",
56
  task="text2text-generation",
57
  huggingfacehub_api_token=HF_TOKEN,
58
- model_kwargs={"temperature": 0.7, "max_new_tokens": 512},
 
59
  )
60
 
 
61
  qa_chain = RetrievalQA.from_chain_type(
62
  llm=llm,
63
  retriever=retriever,
 
55
  repo_id="google/flan-t5-large",
56
  task="text2text-generation",
57
  huggingfacehub_api_token=HF_TOKEN,
58
+ temperature=0.7,
59
+ max_new_tokens=512,
60
  )
61
 
62
+
63
  qa_chain = RetrievalQA.from_chain_type(
64
  llm=llm,
65
  retriever=retriever,