Daksh Chaudhary commited on
Commit
c6e061c
·
verified ·
1 Parent(s): 7af7980

Updated Mistral model

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
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-small-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."""
 
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."""