Hugging Face's logo

Spaces:
Darayut
/
SEALION-v3.5-8B-R-RAG
Sleeping

App Files Community
Darayut commited on
Commit
ddc0c6d
·
verified ·
1 Parent(s): dd69e8f

Update src/simple_rag.py

Browse files
Files changed (1) hide show
  1. src/simple_rag.py +6 -2
src/simple_rag.py CHANGED
@@ -32,8 +32,12 @@ pipe = pipeline(
32
  tokenizer=tokenizer,
33
  )
34
 
35
- DATA_PATH = "./data/"
36
- CHROMA_PATH = "chroma"
 
 
 
 
37
  embedding_model = HuggingFaceEmbeddings(model_name="intfloat/multilingual-e5-base")
38
 
39
  # Generic assistant prompt for dual Khmer/English
 
32
  tokenizer=tokenizer,
33
  )
34
 
35
+ # Use Hugging Face's writable directory
36
+ WRITABLE_DIR = os.environ.get("HF_SPACE_DIR", "/data")
37
+
38
+ DATA_PATH = os.path.join(WRITABLE_DIR, "data")
39
+ CHROMA_PATH = os.path.join(WRITABLE_DIR, "chroma")
40
+
41
  embedding_model = HuggingFaceEmbeddings(model_name="intfloat/multilingual-e5-base")
42
 
43
  # Generic assistant prompt for dual Khmer/English