Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import numpy as np
+import matplotlib.pyplot as plt
 from PIL import Image
 from cellpose import models, io, plot
 
@@ -35,5 +36,8 @@ if __name__ == "__main__":
                 model_path="./cellpose_residual_on_style_on_concatenation_off_fold1_ep_649_cv_0.2834",
                 **model_params
             )
-
-        st.image(preds[0])
+        
+        fig, ax = plt.subplots()
+        ax.imshow(preds[0])
+        
+        st.pyplot(fig)