Update app.py

app.py

@@ -233,7 +233,7 @@ def server(input: Inputs, output: Outputs, session: Session):
             ax.axis("off")  # Hide axis
             return fig
         
-        fig, axes = plt.subplots(1, 4, figsize=(15, 5))
+        fig, axes = plt.subplots(1, 4, figsize=(20, 20), constrained_layout=True)
         
         # Extract the image data
         #image_data = image.cpu().detach().numpy()