hcs commited on
Commit
70c2d4f
·
1 Parent(s): 63d65b3

Add application file

Browse files
Files changed (1) hide show
  1. core.py +1 -1
core.py CHANGED
@@ -39,4 +39,4 @@ def fun(img_path):
39
  with torch.no_grad():
40
  img = img.unsqueeze(0).to(device)
41
  output = net(img).squeeze(1).cpu().numpy()[0]
42
- return "%.1f" % (output * 20)
 
39
  with torch.no_grad():
40
  img = img.unsqueeze(0).to(device)
41
  output = net(img).squeeze(1).cpu().numpy()[0]
42
+ return min(100, "%.1f" % (output * 20) + 15)