fixed a bug with input shape for species model

Shlomigreen · Jun 16, 2021 · dbb4468 · dbb4468
1 parent 5923870
commit dbb4468
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/app.py b/app.py
@@ -111,7 +111,7 @@ def main():
         image = Image.open(uploaded_file)
 
         # Convert to numpy array with proper dimensions for prediction
-        img_array = convert_to_predictable(image)
+        img_array = convert_to_predictable(image, model.input_shape[1:3])
 
         # predict species
         species = predict_species(model, img_array)