Spaces:

anirudh-sub
/

stationary

Runtime error

anirudh-sub commited on Jul 28, 2023

Commit

8061451

1 Parent(s): e969fd6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import gradio as gr
 learn = load_learner('model_v1_75_percent.pkl')
-categories = ('calculator', 'binder clip', 'crayon', 'tape', 'glue', 'marker', 'pencil', 'pen', 'paper', 'eraser')
 def classify_image(img):
   pred, idx, probs = learn.predict(img)

 learn = load_learner('model_v1_75_percent.pkl')
+categories = ('binder clip', 'calculator', 'crayon', 'eraser', 'glue', 'marker', 'pen', 'pencil', 'paper', 'tape')
 def classify_image(img):
   pred, idx, probs = learn.predict(img)