Update app.py

app.py

@@ -1,8 +1,8 @@
 import numpy as np
 import base64
 import cv2
-import streamlit as st
 from fastai.vision.all import *
+import streamlit as st
 st.markdown('<h1 style="color:black;">BDI Image classification model</h1>', unsafe_allow_html=True)
 
 @st.cache(allow_output_mutation=True)
@@ -15,8 +15,8 @@ def get_base64_of_bin_file(bin_file):
 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)
+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')