Update app.py

app.py

@@ -38,7 +38,7 @@ def predict(img):
         output = torch.squeeze(model(img.to(device))).cpu()
         predict = torch.softmax(output, dim=0)
         predict_cla = torch.argmax(predict).numpy()
-    print_res = "class: {}                    prob: {:.3}%".format(class_indict[str(predict_cla)],
+    print_res = "class: {}    \n prob: {:.3}".format(class_indict[str(predict_cla)],
                                                  predict[predict_cla].numpy())
     return print_res