Spaces:
Build error
Build error
Update app.py
Browse files
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 |
-
|
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]
|