Fixing HF Token

app.py

@@ -2,6 +2,7 @@
 
 import os
 import asyncio
+from huggingface_hub import HfFolder
 
 import gradio as gr
 from synthgenai import DatasetConfig, DatasetGeneratorConfig, LLMConfig, InstructionDatasetGenerator, PreferenceDatasetGenerator,RawDatasetGenerator,SentimentAnalysisDatasetGenerator, SummarizationDatasetGenerator, TextClassificationDatasetGenerator
@@ -30,6 +31,18 @@ def stop_generation():
     """
     stop_event.set()
 
+def get_hf_token():
+    """
+    Retrieve the Hugging Face token from the huggingface_hub.
+
+    Returns:
+        str: The Hugging Face token.
+    """
+    token = HfFolder.get_token()
+    if not token:
+        raise ValueError("Hugging Face token not found. Please login using the LoginButton.")
+    return token
+
 def generate_synthetic_dataset(
     llm_model,
     temperature,
@@ -65,6 +78,9 @@ def generate_synthetic_dataset(
         str: A message indicating the result of the dataset generation.
     """
 
+    hf_token = get_hf_token()
+    os.environ["HF_TOKEN"] = hf_token
+
     for var in llm_env_vars.split(","):
         if "=" in var:
             key, value = var.split("=", 1)
@@ -141,7 +157,11 @@ def ui_main():
     """
     with gr.Blocks(
         title="SynthGenAI Dataset Generator",
-        css="footer {visibility: hidden}",
+        css="""
+            .gradio-container .gr-block {
+                margin-bottom: 10px;
+            }
+        """,
         theme="ParityError/Interstellar",
     ) as demo:
         gr.Markdown(