Osnly commited on
Commit
b5b1133
·
verified ·
1 Parent(s): 0f569ec

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +3 -4
src/streamlit_app.py CHANGED
@@ -22,10 +22,9 @@ os.makedirs(charts_dir, exist_ok=True)
22
 
23
  # Authenticate and load tokenizer to check access
24
  hf_token = os.environ.get("HUGGINGFACE_TOKEN")
25
- tokenizer = AutoTokenizer.from_pretrained(
26
- "google/gemma-3n-E4B-it",
27
- token=hf_token
28
- )
29
 
30
  st.set_page_config(page_title="Smart Data Cleaning Agent", layout="wide")
31
  st.title("🧠 Smart Data Cleaning Agent")
 
22
 
23
  # Authenticate and load tokenizer to check access
24
  hf_token = os.environ.get("HUGGINGFACE_TOKEN")
25
+ cache_dir = "/tmp/hf_cache"
26
+ tokenizer = AutoTokenizer.from_pretrained("google/gemma-3n-E4B-it", token=hf_token, cache_dir=cache_dir)
27
+
 
28
 
29
  st.set_page_config(page_title="Smart Data Cleaning Agent", layout="wide")
30
  st.title("🧠 Smart Data Cleaning Agent")