Upload 4 files

app.py

@@ -0,0 +1,64 @@
+import os
+import streamlit as st
+import numpy as np
+import PIL.Image
+#from PIL import Image
+from fastai.vision.all import * 
+import pathlib
+
+import matplotlib.pyplot as plt
+
+
+temp = pathlib.PosixPath
+pathlib.PosixPath = pathlib.WindowsPath
+
+model = load_learner('ksl_model.pkl')
+
+def predict(image_path):
+    # load the image and convert into
+    # numpy array
+    #image= Image.open(image)
+    # image = Image.open(image)
+    # PIL images into NumPy arrays
+    pred_label= model.predict(image_path)
+
+    return pred_label
+
+
+def show_likelihood(pred_label):
+    class_probs = pred_label[2].numpy()
+    classes = ["Temple", "You", "Me", "You", "Friend", "Love", "Enough", "Church","Mosque"]
+    class_labels = [classes[i] for i in range(len(class_probs))]
+    fig = plt.figure(figsize=(10, 10))
+    plt.barh(class_labels, class_probs)
+    plt.ylabel("Class")
+    plt.xlabel("Probability")
+    plt.title("Class Probabilities")
+    plt.xlim(0, 1)
+    plt.ylim(-1, len(class_probs))
+    st.pyplot(fig)
+
+def main():
+    st.set_page_config(page_title="Image Classification App", page_icon=":camera:", layout="wide")
+
+    st.write("# KSL Image Classification App")
+    st.write("This app allows you to upload an image and have it classified by a trained machine learning model.")
+
+    uploaded_file = st.file_uploader("Upload an image", type=["jpg", "png", "jpeg"])
+    if uploaded_file is not None:
+
+        image = PIL.Image.open(uploaded_file)
+
+        image_path = os.path.join("tempDir",uploaded_file.name)
+
+        with open(image_path, "wb") as f: 
+          f.write(uploaded_file.getbuffer())
+
+        st.image(image, caption="Uploaded Image", use_column_width=True)
+        pred_label = predict(image_path)
+        st.write("The image was classified as:", pred_label[0])
+
+        show_likelihood(pred_label)
+
+if __name__ == '__main__':
+    main()

ksl_model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0154c80331d40908a24272d12e6186a50d3e5a8aaa13e7a5232c61e44dd4ef62
+size 87748741

requirements.txt

@@ -0,0 +1,5 @@
+fastai==2.7.10
+matplotlib==3.6.3
+numpy==1.23.5
+Pillow==9.4.0
+streamlit==1.18.1