Testys commited on
Commit
40e40ef
1 Parent(s): 63b58c4

Trying to fix image bug

Browse files
Files changed (1) hide show
  1. main.py +1 -2
main.py CHANGED
@@ -23,8 +23,7 @@ def main():
23
  accept_multiple_files=True)
24
  for in_img in img_input:
25
  if in_img is not None:
26
- image = io.BytesIO(in_img.read())
27
- img = load_image(in_img)
28
  capt = img_caption.get_caption(img)
29
  st.image(image=img, caption=capt, width=250, height=250)
30
  generated_topics = topic_generator.generate_topics(capt)
 
23
  accept_multiple_files=True)
24
  for in_img in img_input:
25
  if in_img is not None:
26
+ img = in_img.read()
 
27
  capt = img_caption.get_caption(img)
28
  st.image(image=img, caption=capt, width=250, height=250)
29
  generated_topics = topic_generator.generate_topics(capt)