frncscp commited on
Commit
612eb22
·
1 Parent(s): a170ddd

Update pages/Entorno de Ejecución.py

Browse files
Files changed (1) hide show
  1. pages/Entorno de Ejecución.py +6 -1
pages/Entorno de Ejecución.py CHANGED
@@ -18,7 +18,7 @@ st.set_page_config(
18
  )
19
 
20
  st.sidebar.write("contact@patacotron.tech")
21
-
22
  cnn, vit, zero_shot, autoencoder, svm, iforest, gan = st.tabs(["CNN", "ViT", "Zero-Shot", "Autoencoder", "OC-SVM", 'iForest', 'GAN'])
23
 
24
  def predict(_model_list, _weights, _img):
@@ -85,6 +85,7 @@ with cnn:
85
  model_dir = os.path.join(DIR, model)
86
  model_name = 'Patacotrón ' + model_dir.split(common_root)[-1].split(common_end)[0]
87
  model_dict[model_name] = model_dir
 
88
  #ultraversions = ['Patacotrón 1.5', 'Patacotrón 1.7', 'Patacotrón 1.8', 'Patacotrón 1.12', 'Patacotrón 1.12.2', 'Patacotrón 1.12.3']
89
  #ultraversions = ['Patacotrón 1.5', 'Patacotrón 1.6', 'Patacotrón 1.12.2', 'Patacotrón 1.8', 'Patacotrón 1.12']#, 'Patacotrón 1.13.20', 'Patacotrón 1.13.38']
90
  #['Patacotrón 1.5', 'Patacotrón 1.6', 'Patacotrón 1.7', 'Patacotrón 1.12'] #
@@ -104,6 +105,8 @@ with cnn:
104
 
105
  threshold = st.slider('¿Cuál va a ser el límite donde se considere patacón? (el valor recomendado es de 75%-80%)', 0, 100, 50, key = 'threshold_convnet')
106
 
 
 
107
  selected_models = []
108
 
109
  # Set the image dimensions
@@ -114,6 +117,8 @@ with cnn:
114
  with col_b:
115
 
116
  uploaded_file = st.file_uploader(key = 'conv_upload', label = 'Sube la imagen a clasificar',type= ['jpg','png', 'jpeg', 'jfif', 'webp', 'heic'])
 
 
117
 
118
  if st.button(key = 'convnet_button', label ='¿Hay un patacón en la imagen?'):
119
  if len(model_choice) < 1:
 
18
  )
19
 
20
  st.sidebar.write("contact@patacotron.tech")
21
+ already_excuted = False
22
  cnn, vit, zero_shot, autoencoder, svm, iforest, gan = st.tabs(["CNN", "ViT", "Zero-Shot", "Autoencoder", "OC-SVM", 'iForest', 'GAN'])
23
 
24
  def predict(_model_list, _weights, _img):
 
85
  model_dir = os.path.join(DIR, model)
86
  model_name = 'Patacotrón ' + model_dir.split(common_root)[-1].split(common_end)[0]
87
  model_dict[model_name] = model_dir
88
+ st.write(model_dict)
89
  #ultraversions = ['Patacotrón 1.5', 'Patacotrón 1.7', 'Patacotrón 1.8', 'Patacotrón 1.12', 'Patacotrón 1.12.2', 'Patacotrón 1.12.3']
90
  #ultraversions = ['Patacotrón 1.5', 'Patacotrón 1.6', 'Patacotrón 1.12.2', 'Patacotrón 1.8', 'Patacotrón 1.12']#, 'Patacotrón 1.13.20', 'Patacotrón 1.13.38']
91
  #['Patacotrón 1.5', 'Patacotrón 1.6', 'Patacotrón 1.7', 'Patacotrón 1.12'] #
 
105
 
106
  threshold = st.slider('¿Cuál va a ser el límite donde se considere patacón? (el valor recomendado es de 75%-80%)', 0, 100, 50, key = 'threshold_convnet')
107
 
108
+
109
+
110
  selected_models = []
111
 
112
  # Set the image dimensions
 
117
  with col_b:
118
 
119
  uploaded_file = st.file_uploader(key = 'conv_upload', label = 'Sube la imagen a clasificar',type= ['jpg','png', 'jpeg', 'jfif', 'webp', 'heic'])
120
+
121
+ st.write(model_choice)
122
 
123
  if st.button(key = 'convnet_button', label ='¿Hay un patacón en la imagen?'):
124
  if len(model_choice) < 1: