Commit
·
6802ee4
1
Parent(s):
0811f98
Last try
Browse files
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)
|