frncscp commited on
Commit
80eca14
1 Parent(s): 79efe44

Update pages/Entorno de Ejecución.py

Browse files
Files changed (1) hide show
  1. pages/Entorno de Ejecución.py +1 -1
pages/Entorno de Ejecución.py CHANGED
@@ -61,7 +61,7 @@ with cnn:
61
  img = tf.image.resize(raw_img,(IMAGE_WIDTH, IMAGE_HEIGHT))
62
  for model, weight in zip(model_list, weights):
63
  y_gorrito += tf.cast(model(tf.expand_dims(img/255., 0)), dtype=tf.float32)*weight
64
- return [y_gorrito / sum(weights, raw_img.numpy()]
65
 
66
  def basic_predict(model_list, weights, img): #for non-supported formats
67
  y_gorrito = 0
 
61
  img = tf.image.resize(raw_img,(IMAGE_WIDTH, IMAGE_HEIGHT))
62
  for model, weight in zip(model_list, weights):
63
  y_gorrito += tf.cast(model(tf.expand_dims(img/255., 0)), dtype=tf.float32)*weight
64
+ return [y_gorrito / sum(weights), raw_img.numpy()]
65
 
66
  def basic_predict(model_list, weights, img): #for non-supported formats
67
  y_gorrito = 0