Update synthetic_generator.py

synthetic_generator.py

@@ -4,6 +4,7 @@ from sklearn.preprocessing import LabelEncoder
 import os
 import json
 import requests
+import streamlit as st
 
 def train_and_generate_synthetic(real_data, schema, output_path):
     """Trains a CTGAN model and generates synthetic data."""
@@ -35,7 +36,7 @@ def train_and_generate_synthetic(real_data, schema, output_path):
 def generate_schema(prompt):
     """Fetches schema from an external API and validates JSON."""
     API_URL = "https://infinitymatter-synthetic-data-generator-srijan.hf.space/run/predict"
-    headers = {"Authorization": f"Bearer hf_token"}  # Add if needed
+    headers = {"Authorization": f"Bearer {st.secrets['hf_token']}"}
 
     try:
         response = requests.post(API_URL, json={"prompt": prompt}, headers=headers)