Update app.py

app.py

@@ -60,19 +60,21 @@ def warmup(model_list=warmup_test, model_inference_endpoints=inference_endpoint)
             response = requests.post(API_URL, headers=headers, json=payload)
             return response.json()
         try:
-            query({
+            _ = query({
                 "inputs": "Hello."
             })
         except:
-            continue
-    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)
-    }
+            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
 
 def inference(model_list, question, API_KEY, cot):
     if len(model_list) != 3: