Commit
·
15f2b4c
1
Parent(s):
b625806
Update app.py
Browse files
app.py
CHANGED
@@ -38,8 +38,8 @@ def predict(img):
|
|
38 |
output = torch.squeeze(model(img.to(device))).cpu()
|
39 |
predict = torch.softmax(output, dim=0)
|
40 |
predict_cla = torch.argmax(predict).numpy()
|
41 |
-
print_res = "class: {}
|
42 |
-
predict[predict_cla].numpy()
|
43 |
return print_res
|
44 |
|
45 |
import gradio as gr
|
|
|
38 |
output = torch.squeeze(model(img.to(device))).cpu()
|
39 |
predict = torch.softmax(output, dim=0)
|
40 |
predict_cla = torch.argmax(predict).numpy()
|
41 |
+
print_res = "class: {} prob: {:.3}%".format(class_indict[str(predict_cla)],
|
42 |
+
predict[predict_cla].numpy())
|
43 |
return print_res
|
44 |
|
45 |
import gradio as gr
|