chrissoria commited on
Commit
30c88db
·
1 Parent(s): 74a0980

Add Mistral to free models

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -25,6 +25,7 @@ FREE_MODEL_CHOICES = [
25
  "meta-llama/Llama-4.1-8B-Instruct:novita",
26
  "gemini-2.5-flash",
27
  "gpt-4o",
 
28
  ]
29
 
30
  # Paid models (user provides their own API key)
@@ -117,6 +118,10 @@ def classify_data(spreadsheet_file, spreadsheet_column,
117
  actual_api_key = os.environ.get("GOOGLE_API_KEY", "")
118
  if not actual_api_key:
119
  return None, None, "**Error:** Google API key not configured in Space secrets"
 
 
 
 
120
  else:
121
  actual_api_key = os.environ.get("HF_API_KEY", "")
122
  else:
 
25
  "meta-llama/Llama-4.1-8B-Instruct:novita",
26
  "gemini-2.5-flash",
27
  "gpt-4o",
28
+ "mistral-large-latest",
29
  ]
30
 
31
  # Paid models (user provides their own API key)
 
118
  actual_api_key = os.environ.get("GOOGLE_API_KEY", "")
119
  if not actual_api_key:
120
  return None, None, "**Error:** Google API key not configured in Space secrets"
121
+ elif "mistral" in model.lower():
122
+ actual_api_key = os.environ.get("MISTRAL_API_KEY", "")
123
+ if not actual_api_key:
124
+ return None, None, "**Error:** Mistral API key not configured in Space secrets"
125
  else:
126
  actual_api_key = os.environ.get("HF_API_KEY", "")
127
  else: