Update app.py

app.py

@@ -57,7 +57,7 @@ def plot_prediction(pred):
     with colB: st.pyplot()
 
 # Define function to plot the decomposed prediction
-def plot_prediction(pred):
+def plot_decomposed(pred):
     plt.figure(figsize=(4, 4))
     plt.imshow(pred, origin="lower")
     plt.axis('off')