File size: 1,063 Bytes
66d7526 f338d56 66d7526 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
import streamlit as st
from PIL import Image
import numpy as np
# Designing the interface
st.title("WIT: Image -> Caption App")
# For newline
st.write('\n')
image = Image.open('images/image.png')
show = st.image(image, use_column_width=True)
from model import *
st.sidebar.title("Upload Image")
# Disabling warning
st.set_option('deprecation.showfileUploaderEncoding', False)
# Choose your own image
uploaded_file = st.sidebar.file_uploader(" ", type=['png', 'jpg', 'jpeg'])
if uploaded_file is not None:
image = Image.open(uploaded_file)
show.image(image, 'Uploaded Image', use_column_width=True)
# For newline
st.sidebar.write('\n')
if st.sidebar.button("Click here to get image caption"):
if uploaded_file is None:
st.sidebar.write("Please upload an Image to Classify")
else:
with st.spinner('Generating image caption ...'):
caption = 'dummy caption'
st.success(f'caption: {caption}')
st.sidebar.header("ViT-GPT2 predicts:")
st.sidebar.write(f"caption: {caption}", '\n')
|