Update app.py

app.py

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