m

app.py

@@ -24,7 +24,7 @@ def askcpv(description):
     outputs = model(**encoding)
     sigmoid = torch.nn.Sigmoid()
     probs = sigmoid(outputs.logits.squeeze().cpu())
-    values, indices = torch.topk(probs, k=10)
+    values, indices = torch.topk(probs, k=5)
     # turn predicted id's into actual label names
     # predicted_labels = [id2label[idx] for idx, label in enumerate(predictions) if label == 1.0]
     # return predicted_labels