Update app.py

app.py

@@ -38,9 +38,7 @@ if __name__ == "__main__":
             )
         
         plt.axis('off')
-        fig, ax = plt.subplots()
-        ax.imshow(preds[0])
-        
-        st.pyplot(fig)
-        
-        st.write(len(flows))
+        fig, ax = plt.subplots(1, 2)
+        ax1.imshow(preds[0])
+        ax2.imshow(flows[0])
+        st.pyplot(fig)