unify implementations

app.py

@@ -138,9 +138,8 @@ def run_classifier(image, threshold):
 
     with torch.no_grad():
         logits = model(tensor)
-        probabilities = torch.nn.functional.sigmoid(logits[0])
-        indices = torch.topk(probabilities, 250).indices
-        values = probabilities[indices]
+        probits = torch.nn.functional.sigmoid(logits[0])
+        values, indices = probits.topk(250)
 
     tag_score = dict()
     for i in range(indices.size(0)):