PeterYoung777 commited on
Commit
15f2b4c
·
1 Parent(s): b625806

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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: {} prob: {:.3}%".format(class_indict[str(predict_cla)],
42
- predict[predict_cla].numpy()*100)
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