DavidD003 commited on
Commit
97ef451
·
1 Parent(s): 09de242
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -1,11 +1,13 @@
1
  import gradio as gr
2
  from fastai.vision.all import *
3
  #
4
- learn = load_learner('export.pkl')
5
- labels = learn.dls.vocab
 
6
  def predict(img):
7
  img = PILImage.create(img)
8
  pred,pred_idx,probs = learn.predict(img)
9
- return {labels[i]: float(probs[i]) for i in range(len(labels))}
10
 
11
- gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)), outputs=gr.outputs.Label(num_top_classes=3)).launch(share=True)
 
 
1
  import gradio as gr
2
  from fastai.vision.all import *
3
  #
4
+ #learn = load_learner('export.pkl')
5
+
6
+ #labels = learn.dls.vocab
7
  def predict(img):
8
  img = PILImage.create(img)
9
  pred,pred_idx,probs = learn.predict(img)
10
+ return "Result"#{labels[i]: float(probs[i]) for i in range(len(labels))}
11
 
12
+ #gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)), outputs=gr.outputs.Label(num_top_classes=3)).launch(share=True)
13
+ gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)), outputs=gr.outputs.Label()).launch(share=True)