update for hf

app/filter_review.py

@@ -1,3 +1,8 @@
+import os
+os.environ["TRANSFORMERS_CACHE"] = "/data/cache"
+os.makedirs("/data/cache", exist_ok=True)
+
+
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 import torch.nn.functional as F

app/main.py

@@ -1,10 +1,10 @@
+import os
+os.environ["TRANSFORMERS_CACHE"] = "/data/cache"
+os.makedirs("/data/cache", exist_ok=True)
+
 from fastapi import FastAPI
 from pydantic import BaseModel
 from app.filter_review import filter_review
-import os
-
-os.environ['TRANSFORMERS_CACHE'] = '/tmp/hf_cache'
-os.makedirs('/tmp/hf_cache', exist_ok=True)
 
 app = FastAPI()