Update app.py

app.py

@@ -1,16 +1,18 @@
 from fastai.vision.all import *
 import gradio as gr
-import spaces
 
 learn = load_learner('model.pkl')
 
-@spaces.GPU
+labels = learn.dls.vocab
+
 def predict(img):
     pred, pred_idx, probs = learn.predict(img)
-    return dict(zip(learn.dls.vocab, map(float, probs)))
+    return {labels[i]: float(probs[i]) for i in range(len(labels))}
 
 gr.Interface(
     fn=predict,
     inputs=gr.Image(type='pil'),
-    outputs=gr.Label(num_top_classes=3)
-).launch()
+    outputs=gr.Label(num_top_classes=3),
+    title="Board Classifier",
+    description="Classifies Arduino, ESP32, and Raspberry Pi boards from a photo.",
+).launch()