Update app.py

app.py

@@ -33,13 +33,16 @@ def classify_image(inp):
   #confidences = model(inp)
 
   preds = model(inp).data[0]
-  means = preds.mean(dim=0, keepdim=True)
-  stds = preds.std(dim=0, keepdim=True)
-  preds = 10 * (preds - means) / stds
-  #preds = nnf.normalize(model(inp).data[0], dim=0)
-  preds = nnf.softmax(preds, dim=0)
+  # means = preds.mean(dim=0, keepdim=True)
+  # stds = preds.std(dim=0, keepdim=True)
+  # preds = 10 * (preds - means) / stds
+  # #preds = nnf.normalize(model(inp).data[0], dim=0)
+  # preds = nnf.softmax(preds, dim=0)
   preds = [pred.cpu() for pred in preds]
   preds = [float(pred.detach()) for pred in preds]
+  p_min = min(preds)
+  p_max = max(preds)
+  preds = [(pred+p_min)/(p_max+p_min) for pred in preds]
   print(pd.Series(preds).describe())
 
   #confidences_dict = {classes[i]: float(confidences.data[0][i]) for i in range(len(confidences.data[0]))}