add post-processing

app.py

@@ -29,7 +29,7 @@ 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, '\nSelected Image', use_column_width=True)
+show.image(image, '\n\nSelected Image', use_column_width=True)
 
 # For newline
 st.sidebar.write('\n')

model.py

@@ -53,6 +53,7 @@ def predict(image):
     token_ids = np.array(generation.sequences)[0]
     caption = tokenizer.decode(token_ids)
     caption = caption.replace('<s>', '').replace('</s>', '').replace('<pad>', '')
+    caption.replace("à l'arrière-plan", '').("Une photo noire et blanche d'", '').replace("en arrière-plan", '')
 
     return caption