ktrapeznikov commited on
Commit
ef3d628
·
1 Parent(s): 0412a1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -12
app.py CHANGED
@@ -73,18 +73,6 @@ def get_connotations(emotion, vocab):
73
  vocab_mask = torch.from_numpy(vocab.isin(word_set).values)
74
  return word_set, vocab_mask
75
 
76
- @st.cache(allow_output_mutation = True)
77
- def get_articles():
78
- article = pd.read_csv("/proj/semafor/datasets/all-the-news-2-1.csv", nrows=10000).dropna(subset =["title","article"])
79
- return article
80
-
81
-
82
- @st.cache(allow_output_mutation = True)
83
- def search_articles(keyword,article):
84
- mask = article.title.str.contains(keyword,case=False)
85
- temp = article.loc[mask].sample().iloc[0]
86
- return temp["title"] + " "+ temp["article"]
87
-
88
  @st.cache(allow_output_mutation = True)
89
  def get_model():
90
  tokenizer = RobertaTokenizerFast.from_pretrained('roberta-base')
 
73
  vocab_mask = torch.from_numpy(vocab.isin(word_set).values)
74
  return word_set, vocab_mask
75
 
 
 
 
 
 
 
 
 
 
 
 
 
76
  @st.cache(allow_output_mutation = True)
77
  def get_model():
78
  tokenizer = RobertaTokenizerFast.from_pretrained('roberta-base')