binqiangliu commited on
Commit
206c966
·
1 Parent(s): f58ec9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -35,7 +35,7 @@ def process_documents_and_query(pdf_files, question):
35
  embed_model = LangchainEmbedding(HuggingFaceEmbeddings(model_name=model_name))
36
 
37
  llm = HuggingFaceHub(repo_id=repo_id,
38
- model_kwargs={"min_length": 1024, "max_new_tokens": 5632,
39
  "do_sample": True, "temperature": 0.1,
40
  "top_k": 50, "top_p": 0.95, "eos_token_id": 49155})
41
 
@@ -64,4 +64,4 @@ gr_interface = gr.Interface(
64
  allow_flagging="never"
65
  )
66
 
67
- gr_interface.launch()
 
35
  embed_model = LangchainEmbedding(HuggingFaceEmbeddings(model_name=model_name))
36
 
37
  llm = HuggingFaceHub(repo_id=repo_id,
38
+ model_kwargs={"min_length": 512, "max_new_tokens": 5632,
39
  "do_sample": True, "temperature": 0.1,
40
  "top_k": 50, "top_p": 0.95, "eos_token_id": 49155})
41
 
 
64
  allow_flagging="never"
65
  )
66
 
67
+ gr_interface.launch()