Update app.py

app.py

@@ -15,8 +15,11 @@ def greet(image):
     prediction = model.predict(image.reshape(1,224,448,image.shape[2]))
     prediction = tf.squeeze(tf.round(prediction))
 
+    prediction = np.argmax(prediction,axis=2)
+
     print(prediction.shape)
     
+    
     #print(model.summary())
     return "Hello "