Update app.py
Browse files
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 |
-
|
| 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
|