Resize Image Dynamically

app.py

@@ -166,7 +166,7 @@ with col2:
         img2=st.session_state["output_2"],
         label1='Raw Diagram',
         label2='Inference Prediction',
-        width=1024,
+        width=col2.width,
         starting_position=50,
         show_labels=True,
         make_responsive=True,