Models usage edited

app.py

@@ -6,21 +6,22 @@ import pandas as pd
 from service_dops_api.dops_config import ServiceDopsConfig
 from service_dops_api.dops_classifier import DopsClassifier
 HF_TOKEN = os.getenv('HF_TOKEN')
-def categoriser_predict(input_text):
-    tokenizer = BertTokenizer.from_pretrained("warleagle/service_name_categorizer",
+tokenizer_cat = BertTokenizer.from_pretrained("warleagle/service_name_categorizer",
+                                          token=HF_TOKEN)
+model_cat = BertForSequenceClassification.from_pretrained('warleagle/service_name_categorizer',token=HF_TOKEN)
+
+tokenizer_spec = BertTokenizer.from_pretrained("warleagle/specialists_categorizer_model",
                                           token=HF_TOKEN)
-    model = BertForSequenceClassification.from_pretrained('warleagle/service_name_categorizer',token=HF_TOKEN)
-    clf = pipeline("text-classification", model=model, tokenizer=tokenizer)
+model_spec = BertForSequenceClassification.from_pretrained('warleagle/specialists_categorizer_model',token=HF_TOKEN)
+def categoriser_predict(input_text):
+    clf = pipeline("text-classification", model=model_cat, tokenizer=tokenizer_cat)
     predictions = clf(input_text)
     numeric_label = int(predictions[0]['label'].split("_")[1])
     id2label = pd.read_pickle('id2label_service_categoriser.pickle')
     text_label = id2label[numeric_label]
     return text_label
 def doctor_spec_predict(input_text):
-    tokenizer = BertTokenizer.from_pretrained("warleagle/specialists_categorizer_model",
-                                          token=HF_TOKEN)
-    model = BertForSequenceClassification.from_pretrained('warleagle/specialists_categorizer_model',token=HF_TOKEN)
-    clf = pipeline("text-classification", model=model, tokenizer=tokenizer)
+    clf = pipeline("text-classification", model=model_spec, tokenizer=tokenizer_spec)
     predictions = clf(input_text)
     numeric_label = int(predictions[0]['label'].split("_")[1])
     id2label = pd.read_pickle('id2label_spec_categoriser.pickle')