app

app.py

@@ -63,13 +63,6 @@ def main():
 
     st.header("Try it out!")
 
-    MODEL = "vivianhuang88/bert_twitter_hashtag"
-    print(MODEL)
-    fill_mask = pipeline("fill-mask", model=MODEL, tokenizer=MODEL)
-    print(fill_mask)
-    tokenizer = BertTokenizer.from_pretrained(MODEL, additional_special_tokens=trending_topics)
-    print(tokenizer)
-
     texts = ['Bruce has an electric guitar set in [MASK]. ', \
     'The Batman: Genesis special feature is a must watch. [MASK] '\
     'I don’t understand the need to exaggerate [MASK] Michelle Obama’s facial expressions. ', \
@@ -77,6 +70,13 @@ def main():
      "Evans Chebet ran mile 22 in 4 minutes and 27 seconds to take the men's [MASK] crown 💨"]
 
     selected_text = st.selectbox('Select a text',(texts))
+
+    MODEL = "vivianhuang88/bert_twitter_hashtag"
+    print(MODEL)
+    fill_mask = pipeline("fill-mask", model=MODEL, tokenizer=MODEL)
+    print(fill_mask)
+    tokenizer = BertTokenizer.from_pretrained(MODEL, additional_special_tokens=trending_topics)
+    print(tokenizer)
     candidates = fill_mask(selected_text, targets = trending_topics)
     print(candidates)
     hashtags = get_hashtags(selected_text, candidates, tokenizer)