Update app.py

app.py

@@ -1,27 +1,22 @@
-import numpy as np
-import base64
 import cv2
-from fastai.vision.all import *
 import streamlit as st
+from fastai.vision.all import *
 st.markdown('<h1 style="color:black;">BDI Image classification model</h1>', unsafe_allow_html=True)
 
+
 @st.cache(allow_output_mutation=True)
-def get_base64_of_bin_file(bin_file):
-    with open(bin_file, 'rb') as f:
-        data = f.read()
-    return base64.b64encode(data).decode()
+upload= st.file_uploader('Insert image for classification', type=['png','jpg'])
+c1, c2= st.columns(2)
+if upload is not None:
+  im= Image.open(upload))
+
 
 
-learn = load_learner('export (5).pkl')
 
-categories = ('Dog Destroying Stuff','Neutral')
-upload = st.file_uploader('Insert image for classification', type=['png','jpg'])
-c1, c2 = st.columns(2)
-if upload is not None:
-  im= Image.open(upload)
   c1.header('Input Image')
   c1.image(im)
   pred, idx, probs = learn.predict(im)
+
   c2.header('Output')
   c2.subheader('Predicted class :')
-  c2.write(pred)
+  c2.write(pred)