Cartinoe5930 commited on
Commit
2aa8f92
·
1 Parent(s): 2887740

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = ["llama", "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"]
 
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"]