Spaces:

rsalman
/

fai_bear_classifier

Runtime error

raahimsal commited on Feb 2, 2024

Commit

6b3bad6

1 Parent(s): cdf1691

+ req

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,17 +2,17 @@ import gradio as gr
 from fastai.vision.widgets import *
 from fastbook import *
-learn = load_learner(r"fastbook/export.pkl")
-categories = ('Grizzly','Black','Teddy')
 def classify_images(img):
     pred,idx,probs = learn.predict(img)
     return dict(zip(categories, map(float, probs)))
-image = gr.inputs.Image(shape=(192,192))
-label = gr.outputs.Label()
-examples = ['fastbook/bears/black/0d147856-6c48-449c-80a1-8ea58587d207.jpg']
 def greet(name):

 from fastai.vision.widgets import *
 from fastbook import *
+learn = load_learner("/Users/rs/dl-prep/fastbook/export.pkl")
+categories = ('Black','Grizzly','Teddy')
 def classify_images(img):
     pred,idx,probs = learn.predict(img)
     return dict(zip(categories, map(float, probs)))
+image = gr.Image(height=192,width=192)
+label = gr.Label()
+examples = ['/Users/rs/dl-prep/fastbook/bears/black/0b439e6e-4808-4e98-aa07-9090d3fc2237.jpg', "/Users/rs/dl-prep/fastbook/bears/black/0d147856-6c48-449c-80a1-8ea58587d207.jpg","/Users/rs/dl-prep/fastbook/bears/teddy/0e6b5384-bff6-4ee2-a7bf-77c9c566a4dc.jpg"]
 def greet(name):