minor bug fix

app.py

@@ -7,8 +7,6 @@ import os
 from dotenv import load_dotenv
 
 load_dotenv()
-os.environ['TRANSFORMERS_CACHE'] = '/temp/cache/'
-os.environ['HF_HOME'] = '/temp/cache/'
 
 app = FastAPI()
 app.add_middleware(

helper.py

@@ -13,8 +13,8 @@ ranker_model_name = "sentence-transformers/all-MiniLM-L6-v2"
 class KeyphraseExtractionPipeline(TokenClassificationPipeline):
     def __init__(self, model, *args, **kwargs):
         super().__init__(
-            model=AutoModelForTokenClassification.from_pretrained(model, cache_dir='/temp/cache/'),
-            tokenizer=AutoTokenizer.from_pretrained(model, cache_dir='/temp/cache/'),
+            model=AutoModelForTokenClassification.from_pretrained(model),
+            tokenizer=AutoTokenizer.from_pretrained(model),
             *args,
             **kwargs
         )
@@ -31,10 +31,9 @@ def init_pipeline() :
         summarizer_model_name,
         device_map="cuda",
         torch_dtype=torch.float16,
-        trust_remote_code=True,
-        cache_dir='/temp/cache/'
+        trust_remote_code=True
     )
-    summarizer_tokenizer = AutoTokenizer.from_pretrained(summarizer_model_name, cache_dir='/temp/cache/')
+    summarizer_tokenizer = AutoTokenizer.from_pretrained(summarizer_model_name)
     
     feature_extractor_model = KeyphraseExtractionPipeline(model=feature_extractor_model_name)