Anders Dahlen commited on
Commit
f3433e8
·
1 Parent(s): ed2a592

Importing all fastai stuff

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import gradio as gr
 
2
 
3
 
4
  learn = load_learner('model.pkl')
@@ -8,4 +9,4 @@ def predict(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 import *
3
 
4
 
5
  learn = load_learner('model.pkl')
 
9
  pred,pred_idx,probs = learn.predict(img)
10
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
11
 
12
+ gr.Interface(fn=predict, inputs=gr.inputs.Image(sgi