Update app.py

app.py

@@ -15,7 +15,7 @@ model = load_model("CADET.hdf5")
 
 # Define function to plot the uploaded image
 def plot_image(image_array, scale):
-    st.set_plot_config(plt, figsize=(4, 4))
+    # st.set_plot_config(plt, figsize=(4, 4))
     plt.figure(figsize=(4, 4))
     # plt.subplot(1, 2, 1)
     x0 = image_array.shape[0] // 2 - scale * 128 / 64
@@ -26,7 +26,7 @@ def plot_image(image_array, scale):
 
 # Define function to plot the prediction
 def plot_prediction(image_array, pred):
-    st.set_plot_config(plt, figsize=(8, 4))
+    # st.set_plot_config(plt, figsize=(8, 4))
     plt.figure(figsize=(8, 4))
     plt.subplot(1, 2, 1)
     plt.imshow(image_array, origin="lower")