adorkin commited on
Commit
cc27541
1 Parent(s): e5b30dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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)