henriquemcruz commited on
Commit
8a8850c
1 Parent(s): cb39feb

Update model to 4 spreadsheets

Browse files
Files changed (3) hide show
  1. .DS_Store +0 -0
  2. app.py +1 -1
  3. model.pkl +2 -2
.DS_Store ADDED
Binary file (6.15 kB). View file
 
app.py CHANGED
@@ -2,7 +2,7 @@ from fastai.vision.all import *
2
  import gradio as gr
3
  learn = load_learner('model.pkl')
4
 
5
- categories = ('Rows','Sheets')
6
 
7
  def classify_image(img):
8
  pred,idx,probs = learn.predict(img)
 
2
  import gradio as gr
3
  learn = load_learner('model.pkl')
4
 
5
+ categories = ('Rows','Sheets','Excel','Numbers')
6
 
7
  def classify_image(img):
8
  pred,idx,probs = learn.predict(img)
model.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f255bcfdc55b531ac983944224087a404e6a0e9b53ef5f5e3aae2e013433502e
3
- size 46954927
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:15487b23b52202cc1f77ad7dc82a47cbd484513b230e83a71be6095f7e4ef908
3
+ size 46959471