Update app.py

app.py

@@ -22,7 +22,7 @@ def plot_image(image_array, scale):
     plt.imshow(image_array, origin="lower")
     plt.gca().add_patch(Rectangle((x0, x0), scale*128, scale*128, linewidth=1, edgecolor='w', facecolor='none'))
     plt.axis('off')
-    st.pyplot()
+    st.pyplot(width=400)
 
 # Define function to plot the prediction
 def plot_prediction(image_array, pred):
@@ -35,7 +35,7 @@ def plot_prediction(image_array, pred):
     plt.subplot(1, 2, 2)
     plt.imshow(pred, origin="lower")
     plt.axis('off')
-    st.pyplot()
+    st.pyplot(width=800)
 
 def cut(data0, wcs0, scale=1):
     shape = data0.shape[0]