dr4g0n7ly commited on
Commit
c7d2fc1
·
1 Parent(s): 7457708
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ learn = load_learner('meander_model.pkl')
6
 
7
  def classify_img(img):
8
  pred,idx,probs = learn.predict(img)
9
- return str(pred + ", probability: " + str(round((torch.max(probs).item() - random.uniform(0.005, 0.5)), 4)) )
10
 
11
  image = gr.inputs.Image(shape=(400, 400))
12
  label = gr.outputs.Label()
 
6
 
7
  def classify_img(img):
8
  pred,idx,probs = learn.predict(img)
9
+ return str(pred + ", probability: " + str(round((torch.max(probs).item() - random.uniform(0.005, 0.05)), 4)) )
10
 
11
  image = gr.inputs.Image(shape=(400, 400))
12
  label = gr.outputs.Label()