Spaces:

agrimsomani
/

figure-identifier

Runtime error

agrimsomani commited on Jan 2, 2023

Commit

e93fe0e

•

1 Parent(s): 264fb2d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,8 +3,7 @@ import gradio as gr
 learner = load_learner('model.pkl')
-categories = ('code','linked-list', 'table', 'graph', 'math')
 def classify_img(img):
     pred, idx, probs = learner.predict(img)

 learner = load_learner('model.pkl')
+categories = ('code', 'graph', 'linked-list', 'math', 'table')
 def classify_img(img):
     pred, idx, probs = learner.predict(img)