Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -28,6 +28,7 @@ def get_top_emojis(text, tokenizer, model, top_n=TOP_N):
|
|
28 |
def main():
|
29 |
|
30 |
cur_model_name = DEFAULT_MODEL
|
|
|
31 |
|
32 |
tokenizer = AutoTokenizer.from_pretrained(cur_model_name)
|
33 |
model = AutoModelForSequenceClassification.from_pretrained(cur_model_name)
|
@@ -60,6 +61,7 @@ def main():
|
|
60 |
|
61 |
model_name = st.selectbox("Choose a model", models_to_choose)
|
62 |
if model_name != cur_model_name:
|
|
|
63 |
cur_model_name = model_name
|
64 |
tokenizer = AutoTokenizer.from_pretrained(cur_model_name)
|
65 |
model = AutoModelForSequenceClassification.from_pretrained(cur_model_name)
|
|
|
28 |
def main():
|
29 |
|
30 |
cur_model_name = DEFAULT_MODEL
|
31 |
+
print("cur_model", cur_model_name)
|
32 |
|
33 |
tokenizer = AutoTokenizer.from_pretrained(cur_model_name)
|
34 |
model = AutoModelForSequenceClassification.from_pretrained(cur_model_name)
|
|
|
61 |
|
62 |
model_name = st.selectbox("Choose a model", models_to_choose)
|
63 |
if model_name != cur_model_name:
|
64 |
+
print("reloading model")
|
65 |
cur_model_name = model_name
|
66 |
tokenizer = AutoTokenizer.from_pretrained(cur_model_name)
|
67 |
model = AutoModelForSequenceClassification.from_pretrained(cur_model_name)
|