4rtemi5 commited on
Commit
624b826
1 Parent(s): a519b9e

trying to fix silviattis issue

Browse files
Files changed (2) hide show
  1. image2text.py +1 -1
  2. text2image.py +1 -1
image2text.py CHANGED
@@ -80,7 +80,7 @@ def app():
80
  st.bar_chart(chart_data)
81
 
82
  with col2:
83
- st.image(image)
84
  gc.collect()
85
 
86
  elif image_url:
 
80
  st.bar_chart(chart_data)
81
 
82
  with col2:
83
+ st.image(image, use_column_width=True)
84
  gc.collect()
85
 
86
  elif image_url:
text2image.py CHANGED
@@ -189,7 +189,7 @@ def app():
189
  elif dataset_name == "CC":
190
  image_raw = requests.get(image_url, stream=True, allow_redirects=True, headers=headers).raw
191
  image = Image.open(image_raw).convert("RGB")
192
- st.image(image)
193
  break
194
  except (UnidentifiedImageError) as e:
195
  if i == N - 1:
 
189
  elif dataset_name == "CC":
190
  image_raw = requests.get(image_url, stream=True, allow_redirects=True, headers=headers).raw
191
  image = Image.open(image_raw).convert("RGB")
192
+ st.image(image, use_column_width=True)
193
  break
194
  except (UnidentifiedImageError) as e:
195
  if i == N - 1: