|
|
|
|
|
import os |
|
|
from huggingface_hub import snapshot_download |
|
|
import sys |
|
|
|
|
|
|
|
|
|
|
|
os.environ.setdefault("HOME", "/home/user") |
|
|
|
|
|
os.environ.setdefault("STREAMLIT_SERVER_ENABLE_XSRF_PROTECTION", "false") |
|
|
os.environ.setdefault("STREAMLIT_SERVER_ENABLE_CORS", "false") |
|
|
os.environ.setdefault("STREAMLIT_BROWSER_GATHER_USAGE_STATS", "false") |
|
|
os.environ.setdefault("STREAMLIT_SERVER_MAX_UPLOAD_SIZE", "1024") |
|
|
|
|
|
|
|
|
try: |
|
|
cfg_dir = os.path.join(os.environ["HOME"], ".streamlit") |
|
|
os.makedirs(cfg_dir, exist_ok=True) |
|
|
cfg_path = os.path.join(cfg_dir, "config.toml") |
|
|
if not os.path.exists(cfg_path): |
|
|
with open(cfg_path, "w", encoding="utf-8") as f: |
|
|
f.write( |
|
|
"[server]\n" |
|
|
"enableXsrfProtection = false\n" |
|
|
"enableCORS = false\n" |
|
|
"maxUploadSize = 1024\n" |
|
|
"\n[browser]\n" |
|
|
"gatherUsageStats = false\n" |
|
|
) |
|
|
except Exception: |
|
|
pass |
|
|
|
|
|
|
|
|
LOCAL_DIR = snapshot_download( |
|
|
repo_id="zenefil/member", |
|
|
repo_type="space", |
|
|
use_auth_token=os.getenv("HF_TOKEN"), |
|
|
) |
|
|
|
|
|
|
|
|
streamlit_script = os.path.join(LOCAL_DIR, "src", "streamlit_app.py") |
|
|
if not os.path.exists(streamlit_script): |
|
|
print(f"File tidak ditemukan: {streamlit_script}") |
|
|
sys.exit(1) |
|
|
|
|
|
os.execvp( |
|
|
"streamlit", |
|
|
[ |
|
|
"streamlit", |
|
|
"run", |
|
|
streamlit_script, |
|
|
"--server.address", "0.0.0.0", |
|
|
"--server.port", os.getenv("PORT", "7860"), |
|
|
], |
|
|
) |
|
|
|