unify implementations

app.py

@@ -156,7 +156,8 @@ def run_classifier(image, threshold):
     tensor = transform(img).unsqueeze(0)
 
     with torch.no_grad():
-        values, indices = torch.topk(model(tensor)[0], 250)
+        probits = model(tensor)[0]
+        values, indices = probits.topk(250)
 
     tag_score = dict()
     for i in range(indices.size(0)):