Update pages/Entorno de Ejecución.py

pages/Entorno de Ejecución.py

@@ -180,7 +180,7 @@ with vit:
                     #classifiers = [pipeline("image-classification", model= model_dict[model_choice[i]]) for i in range(len(model_choice))]
                     
                     #classifier = pipeline("image-classification", model= model_dict[model_choice[0]])
-                    ##img = preprocess(uploaded_file, module = 'pil')
+                    img = preprocess(uploaded_file, module = 'pil')
 
 
                     #def vit_ensemble(classifier_list, img):
@@ -195,10 +195,9 @@ with vit:
 
                     #y_gorrito = query(uploaded_file.read(), model_choice[0])[1]["score"] 
                     
-                    #for model in model_choice:
-                    #    api_result = query(uploaded_file.read(), model_dict[model]) 
-                    #    y_gorrito += api_result[1]["score"]                   
-                    #y_gorrito /= len(model_choice)
+                    for model in model_choice:
+                        y_gorrito += query(uploaded_file.read(), model_dict[model])                 
+                    y_gorrito /= len(model_choice)
     
                     #st.write("y gorrito calculado", len(model_choice))
                     #classifier = classifier(img) 
@@ -216,7 +215,7 @@ with vit:
                     else:
                         st.error("No se considera que haya un patacón en la imagen")
                     st.caption(f'La probabilidad de que la imagen tenga un patacón es del: {round(float(y_gorrito * 100), 2)}%')  
-                    st.image(uploaded_file.read()) 
+                    st.image(img) 
             else:
                 st.write("Asegúrate de haber subido correctamente la imagen.")