Update pages/Entorno de Ejecución.py

pages/Entorno de Ejecución.py

@@ -182,7 +182,8 @@ with vit:
                     #classifier = pipeline("image-classification", model= model_dict[model_choice[0]])
                     img = preprocess(uploaded_file, module = 'pil')
 
-
+                    models = [model_dict[model] for model in model_choice]
+                    st.write(models)
                     #def vit_ensemble(classifier_list, img):
                     #   y_gorrito = 0
                     #    for classifier in classifier_list:
@@ -192,17 +193,17 @@ with vit:
                     #                y_gorrito += clase["score"]
                     #    return y_gorrito / len(classifier_list)
 
-                    models = [model_choice[i] for i in range(len(model_choice))]
-                    st.write(type(models), models)
+                    #models = [model_dict[i] for i in range(len(model_choice))]
+                    #st.write(type(models), models)
                     
                     #st.write(model_choice)
 
                     y_gorrito = 0
                     #y_gorrito = query(uploaded_file.read(), model_choice[0])[1]["score"] 
                     i = -1
-                    for model in model_choice:
+                    for model in models:
                         i+=1
-                        y_gorrito += query(uploaded_file.read(), model_dict[model_choice[i]])    
+                        y_gorrito += query(uploaded_file.read(), model)    
                     y_gorrito /= i
     
                     #st.write("y gorrito calculado", len(model_choice))