saiteja020 commited on
Commit
81dfbda
·
1 Parent(s): cfd5b9a

To HF Space

Browse files
backend/api/llm_evaluator.py CHANGED
@@ -14,9 +14,8 @@ from dotenv import load_dotenv
14
  load_dotenv()
15
 
16
  # Default Model IDs for the HF Router
17
- DEFAULT_GEMMA_MODEL = os.getenv("HF_GEMMA_MODEL", "google/gemma-2-9b-it")
18
  DEFAULT_QWEN_MODEL = os.getenv("HF_QWEN_MODEL", "Qwen/Qwen2.5-72B-Instruct")
19
- DEFAULT_GPT_MODEL = os.getenv("HF_GPT_MODEL", "openai/gpt-4o-mini")
20
  DEFAULT_FALLBACK_MODEL = "meta-llama/Llama-3.3-70B-Instruct"
21
 
22
  TASK_DESCRIPTIONS = {
@@ -221,10 +220,9 @@ def get_llm_analysis(data: dict, provider: str = "QWEN", model_name: str = None)
221
  try:
222
  if provider == "QWEN":
223
  model_id = model_name or DEFAULT_QWEN_MODEL
224
- elif provider == "OPENAI":
225
- model_id = model_name or DEFAULT_GPT_MODEL
226
  else:
227
- model_id = model_name or DEFAULT_GEMMA_MODEL
 
228
 
229
  return _get_router_analysis(data, model_id)
230
 
 
14
  load_dotenv()
15
 
16
  # Default Model IDs for the HF Router
 
17
  DEFAULT_QWEN_MODEL = os.getenv("HF_QWEN_MODEL", "Qwen/Qwen2.5-72B-Instruct")
18
+ DEFAULT_GPT_MODEL = os.getenv("HF_GPT_MODEL", "openai/gpt-oss-20b")
19
  DEFAULT_FALLBACK_MODEL = "meta-llama/Llama-3.3-70B-Instruct"
20
 
21
  TASK_DESCRIPTIONS = {
 
220
  try:
221
  if provider == "QWEN":
222
  model_id = model_name or DEFAULT_QWEN_MODEL
 
 
223
  else:
224
+ # Default to GPT-OSS 20B (Hugging Face) for any other request
225
+ model_id = model_name or DEFAULT_GPT_MODEL
226
 
227
  return _get_router_analysis(data, model_id)
228
 
frontend/src/pages/Evaluate.jsx CHANGED
@@ -17,8 +17,7 @@ export default function Evaluate() {
17
 
18
  const llmOptions = [
19
  { id: 'QWEN', label: 'Qwen 2.5 72B (Alibaba)', model: 'Qwen/Qwen2.5-72B-Instruct' },
20
- { id: 'OPENAI', label: 'GPT-4o (OpenAI)', model: 'gpt-4o' },
21
- { id: 'HF', label: 'Open Source (Hugging Face)', model: '' },
22
  ]
23
 
24
  const [llmProvider, setLlmProvider] = useState(llmOptions[0].id)
 
17
 
18
  const llmOptions = [
19
  { id: 'QWEN', label: 'Qwen 2.5 72B (Alibaba)', model: 'Qwen/Qwen2.5-72B-Instruct' },
20
+ { id: 'OPENAI', label: 'gpt-oss-20b (Hugging Face)', model: 'openai/gpt-oss-20b' },
 
21
  ]
22
 
23
  const [llmProvider, setLlmProvider] = useState(llmOptions[0].id)