frncscp commited on
Commit
08f6b0c
1 Parent(s): fbb2f5a

Update pages/Entorno de Ejecución.py

Browse files
Files changed (1) hide show
  1. pages/Entorno de Ejecución.py +5 -5
pages/Entorno de Ejecución.py CHANGED
@@ -40,14 +40,14 @@ with cnn:
40
  model_dir = os.path.join(DIR, model)
41
  model_name = 'Patacotrón ' + model_dir.split(common_root)[-1].split(common_end)[0]
42
  model_dict[model_name] = model_dir
43
-
44
- ultraversions = ['Patacotrón 1.5', 'Patacotrón 1.7', 'Patacotrón 1.8', 'Patacotrón 1.12', 'Patacotrón 1.12.2']
45
  ultra_button = st.checkbox('Ultra-Patacotrón (mejores resultados)')
46
  ultra_flag = False
 
47
  if ultra_button:
48
  ultra_flag = True
49
- weight_list = [.5, 1.75, 4, .5, 2]
50
-
51
  # Create a dropdown menu to select the model
52
  model_choice = st.multiselect("Seleccione uno o varios modelos de clasificación", model_dict.keys())
53
  weight_list = []
@@ -87,7 +87,7 @@ with cnn:
87
  uploaded_file = st.file_uploader(label = '',type= ['jpg','png', 'jpeg', 'jfif', 'webp', 'heic'])
88
 
89
  if st.button('¿Hay un patacón en la imagen?'):
90
- if len(selected_models) > 0 and ultra_flag:
91
  st.write('Debe elegir un solo método: Ultra-Patacotrón o selección múltiple.')
92
 
93
  elif uploaded_file is not None:
 
40
  model_dir = os.path.join(DIR, model)
41
  model_name = 'Patacotrón ' + model_dir.split(common_root)[-1].split(common_end)[0]
42
  model_dict[model_name] = model_dir
43
+ 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']
 
44
  ultra_button = st.checkbox('Ultra-Patacotrón (mejores resultados)')
45
  ultra_flag = False
46
+
47
  if ultra_button:
48
  ultra_flag = True
49
+ weight_list = [.5, 1, 3, 1, .8, 1] #[.5, 1.75, 4, .5, 2]
50
+
51
  # Create a dropdown menu to select the model
52
  model_choice = st.multiselect("Seleccione uno o varios modelos de clasificación", model_dict.keys())
53
  weight_list = []
 
87
  uploaded_file = st.file_uploader(label = '',type= ['jpg','png', 'jpeg', 'jfif', 'webp', 'heic'])
88
 
89
  if st.button('¿Hay un patacón en la imagen?'):
90
+ if len(model_choice) > 0 and ultra_flag:
91
  st.write('Debe elegir un solo método: Ultra-Patacotrón o selección múltiple.')
92
 
93
  elif uploaded_file is not None: