phind commited on
Commit
1b66fd4
·
1 Parent(s): 1c6aece
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -39,4 +39,7 @@ with tabs2:
39
  image = Image.open(file_name)
40
  st.image(image, use_column_width=True)
41
  captioner = get_model_image_captioning_pipeline()
42
- st.write(f"generated_text: \n{captioner(image)}")
 
 
 
 
39
  image = Image.open(file_name)
40
  st.image(image, use_column_width=True)
41
  captioner = get_model_image_captioning_pipeline()
42
+ predictions = captioner(image)
43
+ st.write("**generated_text: **")
44
+ for generated_text in predictions:
45
+ st.write(f"\n{generated_text['generated_text']}")