Update pages/Entorno de Ejecución.py

pages/Entorno de Ejecución.py

@@ -16,7 +16,7 @@ with st.sidebar:
     st.write("contact@patacotron.tech")
 
 st.title("Entorno de ejecución")
-cnn, autoencoder, svm, iforest, gan, docc = st.tabs(["CNN", "Autoencoder", "OC-SVM", 'iForest', 'GAN', 'DOCC'])
+cnn, autoencoder, svm, iforest, gan, vit = st.tabs(["CNN", "Autoencoder", "OC-SVM", 'iForest', 'GAN', 'ViT'])
 
 with cnn:
 
@@ -35,8 +35,6 @@ with cnn:
         # Join the path to the models folder
         DIR = os.path.join(root_dir, "models")
         
-        threshold = .75
-        
         models = os.listdir(DIR)
         
         model_dict = dict()
@@ -54,7 +52,10 @@ with cnn:
         
         # Create a dropdown menu to select the model
         model_choice = st.multiselect("Seleccione uno o varios modelos de clasificación", model_dict.keys())
-    
+
+        slide = st.slider('¿Cuál va a ser el límite donde se considere patacón?\nEl predeterminado y con mejores resultados es 75%', 0, 100, 75)
+        threshold = slide/100
+        
         selected_models = []
     
         @tf.function
@@ -89,13 +90,13 @@ with cnn:
                 else:
                     with st.spinner('Cargando predicción...'):
                         selected_models = [load_model(model_dict[model]) for model in model_choice if model not in selected_models]
-                        y_gorrito = predict(selected_models, img)
+                        y_gorrito = round(float(predict(selected_models, img)), 2)
         
                 if y_gorrito >= threshold:
                     st.success("¡Patacón Detectado!")
                 else:
-                    st.error("No se encontró rastro de patacón.")
-                st.caption(f'La probabilidad de que la imagen tenga un patacón es del: {round(float(y_gorrito), 2)*100}%')
+                    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: {y_gorrito * 100}%')
                 st.image(raw_img.numpy())
                 st.caption('Si los resultados no fueron los esperados, por favor, [haz click aquí](https://docs.google.com/forms/d/e/1FAIpQLScH0ZxAV8aSqs7TPYi86u0nkxvQG3iuHCStWNB-BoQnSW2V0g/viewform?usp=sf_link)')
             else:
@@ -109,5 +110,5 @@ with cnn:
         st.write('Próximamente')
     with gan:
         st.write('Próximamente')
-    with docc:
+    with vit:
         st.write('Próximamente')