trying to fix silviattis issue

image2text.py

@@ -80,7 +80,7 @@ def app():
                     st.bar_chart(chart_data)
 
                 with col2:
-                    st.image(image)
+                    st.image(image, use_column_width=True)
         gc.collect()
 
     elif image_url:

text2image.py

@@ -189,7 +189,7 @@ def app():
                 elif dataset_name == "CC":
                     image_raw = requests.get(image_url, stream=True, allow_redirects=True, headers=headers).raw
                     image = Image.open(image_raw).convert("RGB")
-                    st.image(image)
+                    st.image(image, use_column_width=True)
                 break
             except (UnidentifiedImageError) as e:
                 if i == N - 1: