Update pages/Entorno de Ejecución.py

pages/Entorno de Ejecución.py

@@ -23,9 +23,9 @@ cnn, vit, zero_shot, autoencoder, svm, iforest, gan = st.tabs(["CNN", "ViT", "Ze
 @st.experimental_singleton
 def predict(_model_list, _weights, _img): 
     y_gorrito = 0
-    raw_img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
-    img = cv2.resize(img, (IMAGE_WIDTH, IMAGE_HEIGHT))
-    for model, weight in zip(model_list, weights):
+    raw_img = cv2.cvtColor(_img, cv2.COLOR_BGR2RGB)
+    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]