frncscp commited on
Commit
1f2ced2
1 Parent(s): 8973c7e

Update pages/Entorno de Ejecución.py

Browse files
Files changed (1) hide show
  1. pages/Entorno de Ejecución.py +2 -2
pages/Entorno de Ejecución.py CHANGED
@@ -46,7 +46,7 @@ def _predict(_model_list, _img, sklearn = False):
46
 
47
  else:
48
  for model in _model_list:
49
- y_gorrito += tf.cast(model(tf.expand_dims(img/255., 0)), dtype=tf.float32)*weight
50
 
51
  return [y_gorrito / len(_model_list), raw_img]
52
 
@@ -126,7 +126,7 @@ with cnn:
126
  model_dir = os.path.join(DIR, model)
127
  model_name = 'Patacotrón ' + model_dir.split(common_root)[-1].split(common_end)[0]
128
  model_dict[model_name] = model_dir
129
- weight_list = []
130
 
131
  # Create a dropdown menu to select the model
132
  model_choice = st.multiselect("Seleccione uno o varios modelos de clasificación", model_dict.keys())
 
46
 
47
  else:
48
  for model in _model_list:
49
+ y_gorrito += tf.cast(model(tf.expand_dims(img/255., 0)), dtype=tf.float32)
50
 
51
  return [y_gorrito / len(_model_list), raw_img]
52
 
 
126
  model_dir = os.path.join(DIR, model)
127
  model_name = 'Patacotrón ' + model_dir.split(common_root)[-1].split(common_end)[0]
128
  model_dict[model_name] = model_dir
129
+ #weight_list = []
130
 
131
  # Create a dropdown menu to select the model
132
  model_choice = st.multiselect("Seleccione uno o varios modelos de clasificación", model_dict.keys())