Spaces:
Running
Running
Commit
·
30c88db
1
Parent(s):
74a0980
Add Mistral to free models
Browse files
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:
|