Soumen commited on
Commit
c0fd49c
·
1 Parent(s): 750eb2a
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -51,4 +51,8 @@ if choice == 'Detection':
51
  preds = tokenizer.batch_decode(output_ids, skip_special_tokens=True)
52
  preds = [pred.strip() for pred in preds]
53
  return preds
54
- st.success(predict_step(our_image))
 
 
 
 
 
51
  preds = tokenizer.batch_decode(output_ids, skip_special_tokens=True)
52
  preds = [pred.strip() for pred in preds]
53
  return preds
54
+ st.success(predict_step(our_image))
55
+ elif choice == 'About':
56
+ st.subheader("About Image Captioning App")
57
+ st.markdown("Built with Streamlit by [Soumen Sarker](https://soumen-sarker-personal-site.streamlit.app/)")
58
+ st.markdown("Demo applicaton of the following model [credit](https://huggingface.co/nlpconnect/vit-gpt2-image-captioning/)")