galihboy commited on
Commit
87e5b75
·
verified ·
1 Parent(s): 45786d1

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +20 -0
src/streamlit_app.py CHANGED
@@ -1,5 +1,25 @@
1
  import os
 
 
 
 
 
 
 
 
2
  os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
 
 
 
 
 
 
 
 
 
 
 
3
 
4
  # Sekarang baru import streamlit
5
  import streamlit as st
 
1
  import os
2
+ from pathlib import Path
3
+
4
+ # Pastikan semua nulis ke /tmp (selalu writable di HF Spaces)
5
+ os.environ.setdefault("HOME", "/tmp")
6
+ os.environ.setdefault("STREAMLIT_USER_SETTINGS_DIR", "/tmp/.streamlit")
7
+ os.environ.setdefault("HF_HOME", "/tmp/.cache/huggingface")
8
+ os.environ.setdefault("TRANSFORMERS_CACHE", "/tmp/.cache/huggingface/transformers")
9
+ os.environ.setdefault("SENTENCE_TRANSFORMERS_HOME", "/tmp/.cache/sentence-transformers")
10
  os.environ["TOKENIZERS_PARALLELISM"] = "false"
11
+ os.environ["STREAMLIT_BROWSER_GATHERUSAGESTATS"] = "false"
12
+ os.environ["STREAMLIT_SERVER_ADDRESS"] = "0.0.0.0"
13
+ os.environ["STREAMLIT_SERVER_PORT"] = os.environ.get("PORT", "7860")
14
+
15
+ # Buat foldernya
16
+ for p in [
17
+ "/tmp/.streamlit",
18
+ "/tmp/.cache/huggingface",
19
+ "/tmp/.cache/huggingface/transformers",
20
+ "/tmp/.cache/sentence-transformers",
21
+ ]:
22
+ Path(p).mkdir(parents=True, exist_ok=True)
23
 
24
  # Sekarang baru import streamlit
25
  import streamlit as st