Update app.py

app.py

@@ -51,16 +51,16 @@ 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"]
+    for model in model_list:
+        API_URL = model_inference_endpoints[model]["API_URL"]
+        headers = model_inference_endpoints[model]["headers"]
         headers["Authorization"] += HF_TOKEN
 
         def query(payload):
             response = requests.post(API_URL, headers=headers, json=payload)
             return response.json()
         try:
-            output = query({
+            query({
                 "inputs": "Hello."
             })
         except: