etadevosyan commited on
Commit
6802ee4
·
1 Parent(s): 0811f98
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -5,6 +5,7 @@ import pickle
5
  import pandas as pd
6
  from service_dops_api.dops_config import ServiceDopsConfig
7
  from service_dops_api.dops_classifier import DopsClassifier
 
8
  def categoriser_predict(input_text):
9
  tokenizer = BertTokenizer.from_pretrained("warleagle/service_name_categorizer",
10
  token=HF_TOKEN)
 
5
  import pandas as pd
6
  from service_dops_api.dops_config import ServiceDopsConfig
7
  from service_dops_api.dops_classifier import DopsClassifier
8
+ HF_TOKEN = os.getenv('HF_TOKEN')
9
  def categoriser_predict(input_text):
10
  tokenizer = BertTokenizer.from_pretrained("warleagle/service_name_categorizer",
11
  token=HF_TOKEN)