Spaces:
Sleeping
Sleeping
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.
|
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()
|