phind commited on
Commit
fa57ca3
·
1 Parent(s): f4ff359
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -40,7 +40,8 @@ with tabs2:
40
  image = Image.open(file_name)
41
  col1.image(image, use_column_width=True)
42
  captioner = get_model_image_captioning()
43
- predictions = captioner(image)
44
  col2.header("generated caption")
 
 
45
  for generated_text in predictions:
46
  col2.write(f"\n{generated_text['generated_text']}")
 
40
  image = Image.open(file_name)
41
  col1.image(image, use_column_width=True)
42
  captioner = get_model_image_captioning()
 
43
  col2.header("generated caption")
44
+ with col2.spinner("Generating caption..."):
45
+ predictions = captioner(image)
46
  for generated_text in predictions:
47
  col2.write(f"\n{generated_text['generated_text']}")