Update

app.py

@@ -70,7 +70,7 @@ def inference(file_path: str):
     predicted_label = logits.argmax(-1).item()
     print(model.config.id2label[predicted_label])
 
-    TOP_K = 5
+    TOP_K = 12
     # logits = np.squeeze(logits)
     logits = logits.squeeze().numpy()
     indices = np.argsort(logits)[::-1][:TOP_K]