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
|