Update app.py

app.py

@@ -3,6 +3,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import numpy as np
 import torch
 
+TOP_N = 5
 
 def preprocess(text):
     new_text = []
@@ -53,7 +54,6 @@ def main():
     models_to_choose = ["AlekseyDorkin/xlm-roberta-en-ru-emoji"]
 
     BASE_MODEL = st.selectbox("Choose a model", models_to_choose)
-    TOP_N = 5
 
     tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
     model = AutoModelForSequenceClassification.from_pretrained(BASE_MODEL)