galihboy commited on
Commit
c869b02
·
verified ·
1 Parent(s): 1b489a4

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +11 -10
src/streamlit_app.py CHANGED
@@ -1,25 +1,26 @@
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
@@ -147,7 +148,7 @@ if st.button("🔍 Search", type="primary") or query:
147
  # Display as dataframe
148
  if results:
149
  df = pd.DataFrame(results)
150
- st.dataframe(df, use_container_width=True, hide_index=True)
151
 
152
  # Detailed view
153
  st.markdown("---")
 
1
  import os
2
  from pathlib import Path
3
 
4
+ # Pastikan Streamlit & HF cache menulis ke /tmp (selalu writable di 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("SENTENCE_TRANSFORMERS_HOME", "/tmp/.cache/sentence-transformers")
9
+ # TRANSFORMERS_CACHE deprecated; HF memakainya dari HF_HOME -> boleh dihapus
10
+ # os.environ.setdefault("TRANSFORMERS_CACHE", "/tmp/.cache/huggingface/transformers")
11
+
12
  os.environ["TOKENIZERS_PARALLELISM"] = "false"
13
  os.environ["STREAMLIT_BROWSER_GATHERUSAGESTATS"] = "false"
14
  os.environ["STREAMLIT_SERVER_ADDRESS"] = "0.0.0.0"
15
  os.environ["STREAMLIT_SERVER_PORT"] = os.environ.get("PORT", "7860")
16
 
17
+ # Buat folder-foldernya
18
+ for p in ["/tmp/.streamlit", "/tmp/.cache/huggingface", "/tmp/.cache/sentence-transformers"]:
 
 
 
 
 
19
  Path(p).mkdir(parents=True, exist_ok=True)
20
+ # ---- END PATCH ----
21
+
22
+ import streamlit as st
23
+
24
 
25
  # Sekarang baru import streamlit
26
  import streamlit as st
 
148
  # Display as dataframe
149
  if results:
150
  df = pd.DataFrame(results)
151
+ st.dataframe(df, width="stretch", hide_index=True)
152
 
153
  # Detailed view
154
  st.markdown("---")