for876543 commited on
Commit
c5b118c
1 Parent(s): feab5d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -29,10 +29,10 @@ def classify_image(inp):
29
  ##confidences = model(inp)
30
 
31
  preds = model(inp).data[0]
32
- means = preds.mean(dim=0, keepdim=True)
33
- stds = preds.std(dim=0, keepdim=True)
34
- preds = 4 * (preds - means) / stds
35
- #preds = nnf.normalize(model(inp).data[0], dim=0)
36
  preds = nnf.softmax(preds, dim=0)
37
  preds = [pred.cpu() for pred in preds]
38
  preds = [float(pred.detach()) for pred in preds]
29
  ##confidences = model(inp)
30
 
31
  preds = model(inp).data[0]
32
+ #means = preds.mean(dim=0, keepdim=True)
33
+ #stds = preds.std(dim=0, keepdim=True)
34
+ #preds = 4 * (preds - means) / stds
35
+ ##preds = nnf.normalize(model(inp).data[0], dim=0)
36
  preds = nnf.softmax(preds, dim=0)
37
  preds = [pred.cpu() for pred in preds]
38
  preds = [float(pred.detach()) for pred in preds]