Update app.py

app.py

@@ -27,19 +27,16 @@ client_z=[]
 
 
 def load_models(inp):
+    
+    out_box=[]
     print(type(inp))
     print(inp)
     print(models[inp[0]])
     client_z.clear()
-    if inp[0]:
-        client_z.append(InferenceClient(models[inp[0]]))
-    if inp[1]:
-        client_z.append(InferenceClient(models[inp[1]]))
-    if inp[2]:
-        client_z.append(InferenceClient(models[inp[2]]))
-    if inp[3]:
-        client_z.append(InferenceClient(models[inp[3]]))
-    return gr.update(label=models[inp[0]])
+    for z,ea in inp:
+        client_z.append(InferenceClient(models[inp[z]]))
+        out_box.append(gr.update(label=models[inp[z]]))
+    return out_box[0]
 
 
 def format_prompt(message, history):