Update app.py

app.py

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