Update app.py

app.py

@@ -30,7 +30,7 @@ def predict_image(image):
     image = tf.keras.preprocessing.image.load_img(image_bytes, target_size=(256, 256))
 
     img_array = preprocess_image(image)
-    outputs = model(img_array)
+    outputs = model.predict(img_array)
     predictions = tf.nn.softmax(outputs.logits, axis=-1)
     predicted_class = np.argmax(predictions)
     #confidence = float(np.max(predictions))