Update app.py

app.py

@@ -279,7 +279,7 @@ def predict(image):
     annotated_img = output[0]
 
     # Reshape the image to match the PIL Image input shape
-    # print("annotated_img shape before reshape:", annotated_img.shape)
+    print("annotated_img shape before reshape:", annotated_img.shape)
     annotated_img = annotated_img.reshape(640, 640, 3)
     print("annotated_img shape after reshape:", annotated_img.shape)