Xinqi04 commited on
Commit
b617d8a
·
1 Parent(s): 9926f86
Files changed (1) hide show
  1. recheck-api.py +6 -1
recheck-api.py CHANGED
@@ -1,6 +1,12 @@
1
  import joblib
2
  import pickle
3
  import pandas as pd
 
 
 
 
 
 
4
  from sentence_transformers import SentenceTransformer
5
  from fastapi.middleware.cors import CORSMiddleware
6
  import re
@@ -29,7 +35,6 @@ except Exception as e:
29
  raise RuntimeError(f"Could not connect to Supabase: {e}")
30
 
31
  try:
32
- os.environ["HF_HOME"] = "./hf_cache"
33
  model_sbert = SentenceTransformer('naufalihsan/indonesian-sbert-large')
34
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
35
  model_sbert.to(device)
 
1
  import joblib
2
  import pickle
3
  import pandas as pd
4
+
5
+ # Atasi masalah permission dengan cache lokal
6
+ os.environ["HF_HOME"] = "./hf_cache"
7
+ os.environ["TRANSFORMERS_CACHE"] = "./hf_cache/transformers"
8
+ os.environ["HF_DATASETS_CACHE"] = "./hf_cache/datasets"
9
+
10
  from sentence_transformers import SentenceTransformer
11
  from fastapi.middleware.cors import CORSMiddleware
12
  import re
 
35
  raise RuntimeError(f"Could not connect to Supabase: {e}")
36
 
37
  try:
 
38
  model_sbert = SentenceTransformer('naufalihsan/indonesian-sbert-large')
39
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
40
  model_sbert.to(device)