Update app.py

app.py

@@ -52,7 +52,7 @@ def LamBERTa(query):
     outputs = model(input_ids, labels=labels)
     loss, logits = outputs[:2]
     log_probs = torch.softmax(logits, dim=1)
-    values, indices = torch.topk(log_probs, 5, dim=1)
+    values, indices = torch.topk(log_probs, 3, dim=1)
     confidences = {id_article[i.item()] : v.item() for i, v in zip(indices[0], values[0])}
     for art, prob in confidences.items():
         texts.append(