Update app.py

app.py

@@ -6,11 +6,8 @@ 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)
+    model = tf.keras.models.load_model('my_model.keras')
     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)
     pred   = np.argmax(prediction, axis=1)[0]
     return pred