Update app.py

app.py

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