update app.py

app.py

@@ -1 +1,46 @@
+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')