Spaces:
Running
Running
Commit
·
2aa8f92
1
Parent(s):
2887740
Update app.py
Browse files
app.py
CHANGED
@@ -50,7 +50,7 @@ def mmlu_display_question_answer(question, cot, request: gr.Request):
|
|
50 |
|
51 |
return q["agent_response"]["llama"][0], q["agent_response"]["wizardlm"][0], q["agent_response"]["orca"][0], q["summarization"][0], q["agent_response"]["llama"][1], q["agent_response"]["wizardlm"][1], q["agent_response"]["orca"][1], q["summarization"][1], q["agent_response"]["llama"][2], q["agent_response"]["wizardlm"][2], q["agent_response"]["orca"][2]
|
52 |
|
53 |
-
warmup_test = ["
|
54 |
def warmup(model_list=warmup_test, model_inference_endpoints=inference_endpoint):
|
55 |
for model in model_list:
|
56 |
API_URL = model_inference_endpoints[model]["API_URL"]
|
|
|
50 |
|
51 |
return q["agent_response"]["llama"][0], q["agent_response"]["wizardlm"][0], q["agent_response"]["orca"][0], q["summarization"][0], q["agent_response"]["llama"][1], q["agent_response"]["wizardlm"][1], q["agent_response"]["orca"][1], q["summarization"][1], q["agent_response"]["llama"][2], q["agent_response"]["wizardlm"][2], q["agent_response"]["orca"][2]
|
52 |
|
53 |
+
warmup_test = ["llama2", "wizardlm", "orca"]
|
54 |
def warmup(model_list=warmup_test, model_inference_endpoints=inference_endpoint):
|
55 |
for model in model_list:
|
56 |
API_URL = model_inference_endpoints[model]["API_URL"]
|