Update app.py

app.py

@@ -22,7 +22,7 @@ if prompt_from_url:
     prompt = prompt_from_url[0]
     image_bytes = query({"inputs": prompt})
     image = Image.open(io.BytesIO(image_bytes))
-    st.image(image, caption="Generated Image", use_column_width=True)
+    st.image(image, caption="Generated Image", use_container_width=True)
 
     # Provide download link
     img_buffer = io.BytesIO()
@@ -49,7 +49,7 @@ else:
 
             # Display Image
             image = Image.open(io.BytesIO(image_bytes))
-            st.image(image, caption="Generated Image", use_column_width=True)
+            st.image(image, caption="Generated Image", use_container_width=True)
 
             # Convert Image to Bytes for Download
             img_buffer = io.BytesIO()