Shreyas094 commited on
Commit
cbab141
1 Parent(s): 2653adc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -596,9 +596,11 @@ 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_tokens=20000,
600
  temperature=temperature,
601
  stream=True,
 
 
602
  ):
603
  if message.choices and message.choices[0].delta and message.choices[0].delta.content:
604
  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_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