Spaces:
Sleeping
Sleeping
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-
|
| 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 |
-
|
|
|
|
| 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: '
|
| 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)
|