Update app.py

app.py

@@ -37,7 +37,10 @@ if __name__ == "__main__":
                 **model_params
             )
         
+        plt.axis('off')
         fig, ax = plt.subplots()
         ax.imshow(preds[0])
         
         st.pyplot(fig)
+        
+        st.write(len(flows))