File size: 725 Bytes
17d7a1f
400190e
cabf856
911ff9f
7dfc144
 
cabf856
 
911ff9f
cabf856
 
 
911ff9f
cabf856
911ff9f
cabf856
911ff9f
cabf856
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
import os
import sys
import subprocess
import streamlit as st
from huggingface_hub import snapshot_download

@st.cache_resource(show_spinner=False)
def get_private_code():
    local_dir = snapshot_download(
        repo_id=os.environ.get("HF_REPO"),
        token=os.environ.get("HF_TOKEN"),
        repo_type="space"
    )
    with open(os.path.join(local_dir, "app.py"), "r", encoding="utf-8") as f:
        code = f.read()
    return code, local_dir

try:
    code_content, private_dir = get_private_code()
    sys.path.append(private_dir)
    os.environ["PYTHONPATH"] = private_dir
    exec(compile(code_content, "internal_logic", "exec"), globals())
except Exception:
    st.error("Error! Check Repo and Token details.")