Update app.py

app.py

@@ -6,8 +6,9 @@ import tensorflow as tf
 
 def image_mod(image):
     # img = Image.fromarray(image['composite'])
-
+    print(image['composite'])
     model = tf.keras.models.load_model('./my_model.keras')
+    print(2222)
     test_img = np.array(image['composite']).reshape(1, test_img.shape[0], test_img.shape[1], 1)
     print(test_img.shape)
     prediction = model.predict(test_img)