Update app.py

app.py

@@ -52,8 +52,8 @@ def mmlu_display_question_answer(question, cot, request: gr.Request):
 warmup_test = ["llama"]
 def warmup(model_list=warmup_test, model_inference_endpoints=inference_endpoint):
     for i in range(len(model_list)):
-        API_URL = model_inference_endpoints[model_list[i]["API_URL"]]
-        headers = model_inference_endpoints[model_list[i]["headers"]]
+        API_URL = model_inference_endpoints[model_list[i]]["API_URL"]
+        headers = model_inference_endpoints[model_list[i]]["headers"]
         headers["Authorization"] += HF_TOKEN
 
         def query(payload):