Update app.py

app.py

@@ -23,11 +23,7 @@ def get_top_emojis(text, top_n=TOP_N):
     preds = model(**inputs).logits
     scores = torch.nn.functional.softmax(preds, dim=-1).detach().numpy()
     ranking = np.argsort(scores)
-    print(ranking)
     ranking = ranking.squeeze()[::-1][:top_n]
-    print(scores)
-    print(ranking)
-    print(model.config.id2label)
     emojis = [model.config.id2label[i] for i in ranking]
     return emojis
     
@@ -37,14 +33,17 @@ gradio_ui = gr.Interface(
     title="Predicting emojis for tweets",
     description="Enter a tweet to predict emojis",
     inputs=[
-        gr.inputs.Textbox(lines=5, label="Paste a tweet here"),
+        gr.inputs.Textbox(lines=3, label="Paste a tweet here"),
     ],
     outputs=[
-        gr.outputs.Textbox(label=f"Label")
+        gr.outputs.Textbox(label=f"Predicted emojis")
     ],
     examples=[
         ["Awesome!"], ["Круто!"], ["lol"]
     ],
+    enable_queue=True,
+    allow_screenshot=False,
+    allow_flagging=False
 )
 
 gradio_ui.launch(debug=True)