Daksh Chaudhary commited on
Updated Mistral model
Browse files
utils.py
CHANGED
|
@@ -14,7 +14,7 @@ def get_llm():
|
|
| 14 |
if not mistral_api_key:
|
| 15 |
raise ValueError("MISTRAL_API_KEY environment variable not set. Please set it before running the app.")
|
| 16 |
|
| 17 |
-
return MistralAI(api_key=mistral_api_key, model="mistral-
|
| 18 |
|
| 19 |
def download_pdf_from_url(url: str):
|
| 20 |
"""Downloads PDF content from a URL and returns it as a BytesIO stream."""
|
|
|
|
| 14 |
if not mistral_api_key:
|
| 15 |
raise ValueError("MISTRAL_API_KEY environment variable not set. Please set it before running the app.")
|
| 16 |
|
| 17 |
+
return MistralAI(api_key=mistral_api_key, model="mistral-medium-latest", timeout=240)
|
| 18 |
|
| 19 |
def download_pdf_from_url(url: str):
|
| 20 |
"""Downloads PDF content from a URL and returns it as a BytesIO stream."""
|