Update app.py

app.py

@@ -3,6 +3,7 @@ import json
 import requests
 import os
 from model_inference import Inference
+import time
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
 question_selector_map = {}
@@ -58,22 +59,20 @@ def warmup(model_list=warmup_test, model_inference_endpoints=inference_endpoint)
 
         def query(payload):
             return requests.post(API_URL, headers=headers, json=payload)
-        # try:
-        #     query({
-        #         "inputs": "Hello. "
-        #     })
-        # except:
-        if model == model_list[-1]:
-            time.sleep(10)
-            return {
-                options: gr.update(visible=True),
-                inputbox: gr.update(visible=True),
-                submit: gr.update(visible=True),
-                warmup_button: gr.update(visible=False),
-                welcome_message: gr.update(visible=True)
-            }
-        else:
-            continue
+            
+        output = query({
+            "inputs": "Hello. "
+        })
+        
+    time.sleep(10)
+    return {
+        options: gr.update(visible=True),
+        inputbox: gr.update(visible=True),
+        submit: gr.update(visible=True),
+        warmup_button: gr.update(visible=False),
+        welcome_message: gr.update(visible=True),
+        dummy: output
+    }
 
 def inference(model_list, question, API_KEY, cot):
     if len(model_list) != 3: