Update app.py

app.py

@@ -17,14 +17,14 @@ def preprocess(text):
         new_text.append(t)
     return " ".join(new_text)
     
-def get_top_emojis(text, top_n=TOP_N):
+def get_top_emojis(text):
     preprocessed = preprocess(text)
     inputs = tokenizer(preprocessed, return_tensors="pt")
     preds = model(**inputs).logits
     scores = torch.nn.functional.softmax(preds, dim=-1).detach().numpy()
-    sorted_scores = [float(value) for value in np.sort(scores.squeeze())[::-1]][:top_n]
+    sorted_scores = [float(value) for value in np.sort(scores.squeeze())[::-1]]
     ranking = np.argsort(scores)
-    ranking = ranking.squeeze()[::-1][:top_n]
+    ranking = ranking.squeeze()[::-1]
     emojis = [model.config.id2label[i] for i in ranking]
     return dict(zip(emojis, sorted_scores))
 
@@ -37,7 +37,7 @@ gradio_ui = gr.Interface(
         gr.inputs.Textbox(lines=3, label="Paste a tweet here"),
     ],
     outputs=[
-        gr.outputs.Label(type="confidences", label=f"Predicted emojis")
+        gr.outputs.Label(type="confidences", label=f"Predicted emojis", num_top_classes=TOP_N)
     ],
     examples=[
         ["it's pretty depressing when u hit pan on ur favourite highlighter"],