Hugging Face's logo

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

App Files Community
Darayut commited on
Commit
0d61d36
·
verified ·
1 Parent(s): 637e09e

Update src/simple_rag.py

Browse files
Files changed (1) hide show
  1. src/simple_rag.py +5 -2
src/simple_rag.py CHANGED
@@ -46,8 +46,11 @@ pipeline = pipeline(
46
  tokenizer=tokenizer,
47
  )
48
 
49
- DATA_PATH = "./data/"
50
- CHROMA_PATH = "chroma"
 
 
 
51
  embedding_model = HuggingFaceEmbeddings(model_name="intfloat/multilingual-e5-base")
52
 
53
  # Generic assistant prompt for dual Khmer/English
 
46
  tokenizer=tokenizer,
47
  )
48
 
49
+ # Use Hugging Face's writable directory
50
+ WRITABLE_DIR = os.environ.get("HOME", "/app")
51
+
52
+ DATA_PATH = os.path.join(WRITABLE_DIR, "src", "data")
53
+ CHROMA_PATH = os.path.join(WRITABLE_DIR, "src", "chroma")
54
  embedding_model = HuggingFaceEmbeddings(model_name="intfloat/multilingual-e5-base")
55
 
56
  # Generic assistant prompt for dual Khmer/English