Update app.py

app.py

@@ -1,6 +1,7 @@
 import cv2
 import streamlit as st
 from fastai.vision.all import *
+import base64
 st.markdown('<h1 style="color:black;">BDI Image classification model</h1>', unsafe_allow_html=True)
 learn = load_learner('export (5).pkl')
 upload = st.file_uploader('Insert image for classification', type=['png','jpg'])