Update app.py

app.py

@@ -16,6 +16,8 @@ def do_prediction(img):
     img_width_model=model.layers[len(model.layers)-1].output_shape[2]
     n_classes=model.layers[len(model.layers)-1].output_shape[3]
 
+    prnt(img_height_model, img_width_model, n_classes,'didi')
+
     kernel = np.ones((5,5),np.uint8)
     margin = int(0.1 * img_width_model)