update to app

.gitignore

@@ -0,0 +1 @@
+tmp/

app.py

@@ -1,5 +1,7 @@
 import streamlit as st
 from fastai.vision.all import *
+import matplotlib.pyplot as plt
+import pandas as pd
 
 def is_cat(x) : return x[0].isupper()
 
@@ -11,12 +13,19 @@ def classify_image(img):
     return dict(zip(categories,map(float,probs)))
 
 def save_uploaded_file(uploadedfile):
-  path_name = os.path.join("tmp",uploadedfile.name)
+  temp_folder = "tmp"
+  if not os.path.exists(temp_folder):
+     os.makedirs(temp_folder)
+  path_name = os.path.join(temp_folder,uploadedfile.name)
   with open(path_name,"wb") as f:
     f.write(uploadedfile.getbuffer())
-    st.success(f"Saved file :{uploadedfile.name} as {path_name}")
   return path_name
 
+st.write("""
+# Dog vs Cat Classifier
+This is a application that classifies images of dogs vs cats
+""")
+
 upload_image = st.file_uploader("Choose a file")
 
 if upload_image is not None:
@@ -25,4 +34,6 @@ if upload_image is not None:
     st.image(image)
     path_name = save_uploaded_file(upload_image)
     st.write("Prediction Propabilities:")
-    st.write(classify_image(path_name))
+    st.write(classify_image(path_name))
+    preds = pd.DataFrame(classify_image(path_name), index=[0])
+    st.bar_chart(preds)