chrissoria commited on
Commit
06bbe2c
·
1 Parent(s): e0d89c2

Add Perplexity Sonar to free models

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -27,6 +27,7 @@ FREE_MODEL_CHOICES = [
27
  "gpt-4o",
28
  "mistral-medium-latest",
29
  "claude-3-haiku-20240307",
 
30
  ]
31
 
32
  # Paid models (user provides their own API key)
@@ -130,6 +131,10 @@ def classify_data(spreadsheet_file, spreadsheet_column,
130
  actual_api_key = os.environ.get("ANTHROPIC_API_KEY", "")
131
  if not actual_api_key:
132
  return None, None, "**Error:** Anthropic API key not configured in Space secrets"
 
 
 
 
133
  else:
134
  actual_api_key = os.environ.get("HF_API_KEY", "")
135
  else:
 
27
  "gpt-4o",
28
  "mistral-medium-latest",
29
  "claude-3-haiku-20240307",
30
+ "sonar",
31
  ]
32
 
33
  # Paid models (user provides their own API key)
 
131
  actual_api_key = os.environ.get("ANTHROPIC_API_KEY", "")
132
  if not actual_api_key:
133
  return None, None, "**Error:** Anthropic API key not configured in Space secrets"
134
+ elif "sonar" in model.lower():
135
+ actual_api_key = os.environ.get("PERPLEXITY_API_KEY", "")
136
+ if not actual_api_key:
137
+ return None, None, "**Error:** Perplexity API key not configured in Space secrets"
138
  else:
139
  actual_api_key = os.environ.get("HF_API_KEY", "")
140
  else: