Update app.py

app.py

@@ -231,9 +231,10 @@ def server(input: Inputs, output: Outputs, session: Session):
             fig, ax = plt.subplots()
             ax.text(0.5, 0.5, "Upload an image to see predictions", fontsize=12, ha="center")
             ax.axis("off")  # Hide axis
+            plt.tight_layout()
             return fig
         
-        fig, axes = plt.subplots(1, 4, figsize=(20, 20), constrained_layout=True)
+        fig, axes = plt.subplots(1, 4, constrained_layout=True)
         
         # Extract the image data
         #image_data = image.cpu().detach().numpy()