Spaces:

frncscp
/

Patacotron

Running

frncscp commited on May 11, 2023

Commit

faae2ad

1 Parent(s): fd96b98

Update pages/Entorno de Ejecución.py

Files changed (1) hide show

pages/Entorno de Ejecución.py CHANGED Viewed

@@ -27,7 +27,7 @@ def predict(_model_list, _weights, _img):
     img = cv2.resize(_img, (IMAGE_WIDTH, IMAGE_HEIGHT))
     for model, weight in zip(_model_list, _weights):
         y_gorrito += tf.cast(model(tf.expand_dims(img/255., 0)), dtype=tf.float32)*weight
-    return [y_gorrito / sum(weights), raw_img]
 def preprocess(file_uploader, module = 'cv2'): #makes the uploaded image readable
     img = np.frombuffer(uploaded_file.read(), np.uint8)

     img = cv2.resize(_img, (IMAGE_WIDTH, IMAGE_HEIGHT))
     for model, weight in zip(_model_list, _weights):
         y_gorrito += tf.cast(model(tf.expand_dims(img/255., 0)), dtype=tf.float32)*weight
+    return [y_gorrito / sum(_weights), raw_img]
 def preprocess(file_uploader, module = 'cv2'): #makes the uploaded image readable
     img = np.frombuffer(uploaded_file.read(), np.uint8)