Spaces:
Running
Running
Updated
Browse files
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/)")
|