dr4g0n7ly commited on
Commit
810a02e
·
1 Parent(s): 3fb9023
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,5 +1,6 @@
1
  from fastai.vision.all import *
2
  import gradio as gr
 
3
 
4
  learn = load_learner('meander_model.pkl')
5
 
@@ -8,7 +9,7 @@ def classify_img(img):
8
  return str(pred + ", probability: " + str(round(torch.max(probs).item(), 4)) )
9
 
10
  image = gr.inputs.Image(shape=(400, 400))
11
- label = gr.outputs.Label()
12
 
13
  intf = gr.Interface(fn=classify_img, inputs=image, outputs=label)
14
  intf.launch(inline=False)
 
1
  from fastai.vision.all import *
2
  import gradio as gr
3
+ import random
4
 
5
  learn = load_learner('meander_model.pkl')
6
 
 
9
  return str(pred + ", probability: " + str(round(torch.max(probs).item(), 4)) )
10
 
11
  image = gr.inputs.Image(shape=(400, 400))
12
+ label = gr.outputs.Label() - random.randint(0.0008,0.001)
13
 
14
  intf = gr.Interface(fn=classify_img, inputs=image, outputs=label)
15
  intf.launch(inline=False)