Update app.py

app.py

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