Xinqi04 commited on
Commit ·
b617d8a
1
Parent(s): 9926f86
new2
Browse files- 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)
|