Shreyas094 commited on
Commit
ce3d14c
·
verified ·
1 Parent(s): cbab141

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -596,11 +596,12 @@ def get_response_from_pdf(query, model, selected_docs, num_calls=3, temperature=
596
  logging.info(f"API call {i+1}/{num_calls}")
597
  for message in client.chat_completion(
598
  messages=messages,
599
- max_new_tokens=20000,
600
  temperature=temperature,
601
  stream=True,
602
  top_p=0.8,
603
  do_sample=False,
 
604
  ):
605
  if message.choices and message.choices[0].delta and message.choices[0].delta.content:
606
  chunk = message.choices[0].delta.content
 
596
  logging.info(f"API call {i+1}/{num_calls}")
597
  for message in client.chat_completion(
598
  messages=messages,
599
+ max_tokens=20000,
600
  temperature=temperature,
601
  stream=True,
602
  top_p=0.8,
603
  do_sample=False,
604
+ top_k=10,
605
  ):
606
  if message.choices and message.choices[0].delta and message.choices[0].delta.content:
607
  chunk = message.choices[0].delta.content