Update app.py
Browse files
app.py
CHANGED
|
@@ -26,7 +26,7 @@ def get_top_emojis(text, top_n=TOP_N):
|
|
| 26 |
ranking = np.argsort(scores)
|
| 27 |
ranking = ranking.squeeze()[::-1][:top_n]
|
| 28 |
emojis = [model.config.id2label[i] for i in ranking]
|
| 29 |
-
return dict(zip(emojis, sorted_scores)
|
| 30 |
|
| 31 |
|
| 32 |
gradio_ui = gr.Interface(
|
|
|
|
| 26 |
ranking = np.argsort(scores)
|
| 27 |
ranking = ranking.squeeze()[::-1][:top_n]
|
| 28 |
emojis = [model.config.id2label[i] for i in ranking]
|
| 29 |
+
return dict(zip(emojis, sorted_scores))
|
| 30 |
|
| 31 |
|
| 32 |
gradio_ui = gr.Interface(
|