fix model path, added example and class names

app.py

@@ -1,12 +1,18 @@
 import pathlib
 import gradio as gr
-from fastai.vision.all import *
-
 # to solve the mismatch between windows (local) system and huggingspace (linux)
+import platform
 plt = platform.system()
 if plt == 'Linux': pathlib.WindowsPath = pathlib.PosixPath
+elif plt == 'Windows': pathlib.PosixPath = pathlib.WindowsPath
+
+from fastai.vision.all import *
+
+
+
+class_names = ["resistor", "bipolar transistor", "mosfet", "capacitor", "inductor", "wire", "led", "diode", "thermistor", "switch", "battery", "hammer", "screwdriver", "scissors", "wrench", "mallet", "axe"]
 
-learn = load_learner('export.pkl')
+learn = load_learner('model.pkl')
 labels = learn.dls.vocab
 
 def predict(img):
@@ -16,11 +22,12 @@ def predict(img):
 
 iface = gr.Interface(
     fn=predict, 
-    inputs=gr.components.Image(shape=(512, 512)), 
+    inputs=gr.components.Image(shape=(224, 224)), 
     outputs=gr.components.Label(num_top_classes=3),
-    description="Pet Classifier",
-    article="<p style='text-align: center'><a href='google.com' target='_blank'>Blog post</a></p>",
-    live=True,    
+    description="Tool Classifier",
+    article=f"<p style='text-align: center'>{', '.join(class_names)}</p>",
+    examples=[["mosfet.jpg", "screwdriver.jpg", "hammel.png", "led.jpg"]],
+    live=True,
     )
 
 iface.launch(enable_queue=True)