1st working model version

.gitignore

@@ -1 +1,2 @@
 app.ipynb
+data/

RM_classifier.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3330dd6dd282482ace5c92ca1ad88dd498deb6cbcc06b7e4b96ee4eb8629361a
-size 46974945
+oid sha256:c5069aa9d65633684dd85c9663ccf68113c80f88547be1a576e0012dcf2e07c4
+size 46978017

app.py

@@ -1,5 +1,10 @@
 from fastai.vision.all import *
 import gradio as gr
+import os
+
+import pathlib
+temp = pathlib.PosixPath
+pathlib.PosixPath = pathlib.WindowsPath
 
 learner = load_learner('RM_classifier.pkl')
 
@@ -11,7 +16,7 @@ def classify_image(image):
 
 image = gr.inputs.Image(shape=(224,224))
 label = gr.outputs.Label()
-examples = ['Morty.png','Summer.png','Birdperson.jpg','Rick.jpg']
+examples = [os.path.join("./images", img_path) for img_path in os.listdir( './images')]
 
 interface = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples, title='Rick and Morty Character Classifier', description='Upload an image of a Rick and Morty character')
 interface.launch()

requirements.txt

@@ -2,3 +2,4 @@ fastai
 torch
 gradio
 numpy
+os