Mwebrania commited on
Commit
d942b24
·
1 Parent(s): 05f9cb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ from fastai import *
7
  from fastai.vision import *
8
  #from fastbook import *
9
 
10
- learn = load_learner('export.pkl')
11
 
12
  labels = learn.dls.vocab
13
  def predict(img):
@@ -23,7 +23,7 @@ interpretation = 'default'
23
  enable_queue = True
24
 
25
  gr.Interface(fn = predict, inputs = gr.inputs.Image(shape = (512, 512)),
26
- outputs = gr.outputs.Label(num_top_classes = 3),
27
  title = title,
28
  description = description,
29
  article = article,
 
7
  from fastai.vision import *
8
  #from fastbook import *
9
 
10
+ learn = load_learner('clasma.pkl')
11
 
12
  labels = learn.dls.vocab
13
  def predict(img):
 
23
  enable_queue = True
24
 
25
  gr.Interface(fn = predict, inputs = gr.inputs.Image(shape = (512, 512)),
26
+ outputs = gr.outputs.Label(num_top_classes = 8),
27
  title = title,
28
  description = description,
29
  article = article,