mbahrami commited on
Commit
6ea6ff3
1 Parent(s): c990aa6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -22,7 +22,7 @@ def loading_models(model='roberta-base'):
22
  suppress_st_warning=True,
23
  hash_funcs={'tokenizers.Tokenizer': hash_func, 'tokenizers.AddedToken': hash_func})
24
  def infer(text):
25
- # global nlp
26
  return nlp(text+' '+nlp.tokenizer.mask_token)
27
 
28
 
@@ -33,8 +33,6 @@ def sim(predicted_seq, sem_list):
33
  return semantic_model.encode(predicted_seq, convert_to_tensor=True), \
34
  semantic_model.encode(sem_list, convert_to_tensor=True)
35
 
36
-
37
-
38
  @st.cache(allow_output_mutation=True,
39
  suppress_st_warning=True,
40
  hash_funcs={'tokenizers.Tokenizer': hash_func, 'tokenizers.AddedToken': hash_func})
22
  suppress_st_warning=True,
23
  hash_funcs={'tokenizers.Tokenizer': hash_func, 'tokenizers.AddedToken': hash_func})
24
  def infer(text):
25
+ # global nlp
26
  return nlp(text+' '+nlp.tokenizer.mask_token)
27
 
28
 
33
  return semantic_model.encode(predicted_seq, convert_to_tensor=True), \
34
  semantic_model.encode(sem_list, convert_to_tensor=True)
35
 
 
 
36
  @st.cache(allow_output_mutation=True,
37
  suppress_st_warning=True,
38
  hash_funcs={'tokenizers.Tokenizer': hash_func, 'tokenizers.AddedToken': hash_func})