Upload model.py

model.py

@@ -1,10 +1,14 @@
 import os
 from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
 
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
+
+
 HF_TOKEN = os.getenv("HF_TOKEN")
 MODEL_ID = "TypicaAI/magbert-ner"
 
-tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, use_auth_token=HF_TOKEN)
-model = AutoModelForTokenClassification.from_pretrained(MODEL_ID, use_auth_token=HF_TOKEN)
+tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, token=HF_TOKEN)
+model = AutoModelForTokenClassification.from_pretrained(MODEL_ID, token=HF_TOKEN)
+
 
 ner_pipeline = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="first")