Ogghey commited on
Commit
d411917
Β·
verified Β·
1 Parent(s): a7eb20d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -5
app.py CHANGED
@@ -1,18 +1,27 @@
1
  import os
 
2
  from fastapi import FastAPI, Request
3
  from sentence_transformers import SentenceTransformer, util
4
  import torch
5
  import requests
6
 
7
- # βœ… Atur path cache agar tidak kena permission error
8
- os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf"
9
- os.makedirs("/tmp/hf", exist_ok=True)
 
 
10
 
11
- # πŸ” Supabase setup
 
 
 
 
 
 
12
  SUPABASE_URL = "https://olbjfxlclotxtnpjvpfj.supabase.co"
13
  SUPABASE_KEY = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6Im9sYmpmeGxjbG90eHRucGp2cGZqIiwicm9sZSI6ImFub24iLCJpYXQiOjE3NTIyMzYwMDEsImV4cCI6MjA2NzgxMjAwMX0.7q_o5DCFEAAysnWXMChH4MI5qNhIVc4OgpT5JvgYxc0"
14
 
15
- # πŸ”„ Gunakan model yang lebih ringan agar tidak error runtime
16
  model = SentenceTransformer("sentence-transformers/paraphrase-MiniLM-L3-v2")
17
 
18
  app = FastAPI()
 
1
  import os
2
+ import shutil
3
  from fastapi import FastAPI, Request
4
  from sentence_transformers import SentenceTransformer, util
5
  import torch
6
  import requests
7
 
8
+ # πŸ” Paksa semua cache ke path aman di Hugging Face Spaces
9
+ HF_CACHE = "/tmp/hf"
10
+ os.environ["TRANSFORMERS_CACHE"] = HF_CACHE
11
+ os.environ["HF_HOME"] = HF_CACHE
12
+ os.makedirs(HF_CACHE, exist_ok=True)
13
 
14
+ # βœ… Bersihkan cache jika terkunci
15
+ if os.path.exists(f"{HF_CACHE}/models--sentence-transformers--paraphrase-MiniLM-L3-v2.lock"):
16
+ os.remove(f"{HF_CACHE}/models--sentence-transformers--paraphrase-MiniLM-L3-v2.lock")
17
+ if os.path.exists(f"{HF_CACHE}/models--sentence-transformers--paraphrase-MiniLM-L3-v2"):
18
+ shutil.rmtree(f"{HF_CACHE}/models--sentence-transformers--paraphrase-MiniLM-L3-v2", ignore_errors=True)
19
+
20
+ # πŸ” Supabase
21
  SUPABASE_URL = "https://olbjfxlclotxtnpjvpfj.supabase.co"
22
  SUPABASE_KEY = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6Im9sYmpmeGxjbG90eHRucGp2cGZqIiwicm9sZSI6ImFub24iLCJpYXQiOjE3NTIyMzYwMDEsImV4cCI6MjA2NzgxMjAwMX0.7q_o5DCFEAAysnWXMChH4MI5qNhIVc4OgpT5JvgYxc0"
23
 
24
+ # πŸ”„ Gunakan model kecil dan cepat
25
  model = SentenceTransformer("sentence-transformers/paraphrase-MiniLM-L3-v2")
26
 
27
  app = FastAPI()