Spaces:

henriquemcruz
/

spreadsheetclassifier

Build error

henriquemcruz commited on Aug 30, 2022

Commit

48e6643

•

1 Parent(s): e2eccb3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ from fastai.vision.all import *
 import gradio as gr
 learn = load_learner('model3.pkl')
-categories = ('Excel', 'Numbers', 'Rows', 'Sheets')
 def classify_image(img):
         pred,idx,probs = learn.predict(img)

 import gradio as gr
 learn = load_learner('model3.pkl')
+categories = learn.dls.vocab
 def classify_image(img):
         pred,idx,probs = learn.predict(img)