Update app.py

app.py

@@ -170,7 +170,7 @@ def display_result(image: Image.Image, prompt: str, index: int, total: int) -> O
     Returns:
         Annotation data (list of dicts) if annotations were made, otherwise None.
     """
-    st.image(image, caption=f"Result {index + 1}/{total}: {prompt}", use_column_width='always')
+    st.image(image, caption=f"Result {index + 1}/{total}: {prompt}", use_container_width=True)
 
     # Prepare image for download
     buf = BytesIO()