Update app.py

app.py

@@ -239,8 +239,8 @@ def server(input: Inputs, output: Outputs, session: Session):
         # Extract the image data
         #image_data = image.cpu().detach().numpy()
 
-        # Get the dimensions of the original image
-        height, width = image.shape[:2]
+        # Get the dimensions (width and height) of the image
+        width, height = image.size
         
         # Plot the first image on the left
         axes[0].imshow(image)