Osnly commited on
Commit
11a46a2
·
verified ·
1 Parent(s): 6b95a85

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +9 -0
src/streamlit_app.py CHANGED
@@ -11,7 +11,16 @@ from execute import execute_plan
11
  from insight import generate_insights
12
  from visual_insight import generate_visual_plan
13
  from report import ReportBuilder
 
 
 
 
14
 
 
 
 
 
 
15
  st.set_page_config(page_title="Smart Data Cleaning Agent", layout="wide")
16
  st.title("🧠 Smart Data Cleaning Agent")
17
 
 
11
  from insight import generate_insights
12
  from visual_insight import generate_visual_plan
13
  from report import ReportBuilder
14
+ import os
15
+ from transformers import AutoTokenizer
16
+
17
+ token = os.environ.get("HUGGINGFACE_TOKEN")
18
 
19
+ tokenizer = AutoTokenizer.from_pretrained(
20
+ "google/gemma-3n-E4B-it",
21
+ token=token, # Required to access gated models
22
+ use_auth_token=True # Optional in newer transformers
23
+ )
24
  st.set_page_config(page_title="Smart Data Cleaning Agent", layout="wide")
25
  st.title("🧠 Smart Data Cleaning Agent")
26