Cartinoe5930 commited on
Commit
c8005a5
·
1 Parent(s): 9f857bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -49,8 +49,8 @@ def mmlu_display_question_answer(question, cot, request: gr.Request):
49
 
50
  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]
51
 
52
-
53
- def warmup(model_list=model_list, model_inference_endpoints=inference_endpoint):
54
  for i in range(len(model_list)):
55
  API_URL = model_inference_endpoints[model_list[i]["API_URL"]]
56
  headers = model_inference_endpoints[model_list[i]["headers"]]
 
49
 
50
  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]
51
 
52
+ warmup_test = ["llama"]
53
+ def warmup(model_list=warmup_test, model_inference_endpoints=inference_endpoint):
54
  for i in range(len(model_list)):
55
  API_URL = model_inference_endpoints[model_list[i]["API_URL"]]
56
  headers = model_inference_endpoints[model_list[i]["headers"]]