Update app.py

app.py

@@ -1,17 +1,19 @@
 import gradio as gr
 import os
+from datasets import load_dataset
 from huggingface_hub import HfApi, login
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-# NVidia A10G Large (sleep after 1 hour)
+# Run on NVidia A10G Large (sleep after 1 hour)
 
 # Model IDs:
+#
 # google/gemma-2-9b-it
 # meta-llama/Meta-Llama-3-8B-Instruct
 
 # Datasets:
 #
-#
+# gretelai/synthetic_text_to_sql
 
 profile = "bstraehle"
 
@@ -23,6 +25,7 @@ def download_model(model_id):
     return tokenizer
 
 def download_dataset(dataset):
+    ds = load_dataset(dataset)
     return ""
     
 def fine_tune_model():