Spaces:

M4869
/

beauty_prediction

Runtime error

hcs commited on Jul 3, 2023

Commit

6d7b1c4

1 Parent(s): 70c2d4f

Add application file

Files changed (1) hide show

core.py CHANGED Viewed

@@ -39,4 +39,5 @@ def fun(img_path):
     with torch.no_grad():
         img = img.unsqueeze(0).to(device)
         output = net(img).squeeze(1).cpu().numpy()[0]
-        return min(100, "%.1f" % (output * 20) + 15)

     with torch.no_grad():
         img = img.unsqueeze(0).to(device)
         output = net(img).squeeze(1).cpu().numpy()[0]
+        v = float("%.1f" % (output * 20))
+        return min(100, v + 15.0)