use real predict method

app.py

@@ -12,7 +12,7 @@ st.write('\n')
 #show = st.image(image, use_column_width=True)
 #show.image(image, 'Preloaded Image', use_column_width=True)
 
-with st.spinner('Loading ViT-GPT2 model ...'):
+with st.spinner('Loading and compiling ViT-GPT2 model ...'):
 
     from model import *
     st.sidebar.write(f'Vit-GPT2 model loaded :)')
@@ -29,14 +29,14 @@ sample_path = os.path.join(sample_dir, sample_name)
 
 image = Image.open(sample_path)
 show = st.image(image, use_column_width=True)
-show.image(image, 'Selected Image', use_column_width=True)
+show.image(image, '\nSelected Image', use_column_width=True)
 
 # For newline
 st.sidebar.write('\n')
 
 with st.spinner('Generating image caption ...'):
 
-    caption = predict_dummy(image)
+    caption = predict(image)
     image.close()
     st.success(f'caption: {caption}')
 

model.py

@@ -52,6 +52,7 @@ def predict(image):
 
     token_ids = np.array(generation.sequences)[0]
     caption = tokenizer.decode(token_ids)
+    caption = caption.replace('<s>', '').replace('</s>', '').replace('<pad>', '')
 
     return caption