Update app.py

app.py

@@ -5,34 +5,23 @@ import streamlit as st
 from fastai.vision.all import *
 st.markdown('<h1 style="color:black;">BDI Image classification model</h1>', unsafe_allow_html=True)
 
-
-
-learn = load_learner('export (5).pkl')
-categories = ('Dog Destroying Stuff','Neutral')
-
-
-
 @st.cache(allow_output_mutation=True)
-
-
-upload= st.file_uploader('Insert image for classification', type=['png','jpg'])
-c1, c2= st.columns(2)
-
-
-
 def get_base64_of_bin_file(bin_file):
     with open(bin_file, 'rb') as f:
         data = f.read()
     return base64.b64encode(data).decode()
 
 
+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)
-    if pred == 'Dog Destroying Stuff':
-        sound()
+  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)