Update app.py

app.py

@@ -9,9 +9,7 @@ from huggingface_hub import from_pretrained_keras
 
 def greet(image):
     model = from_pretrained_keras("vahidrezanezhad/sbb_binarization")
-    print(np.shape(image))
-    print(type(image))
-    prediction = model.predict(image.reshape(1,image.shape[0],image.shape[1],image.shape[2]))
+    prediction = model.predict(image.reshape(1,224,448,image.shape[2]))
     prediction = tf.squeeze(tf.round(prediction))
 
     print(prediction.shape)