Update app.py

app.py

@@ -1,5 +1,6 @@
 import os, torch, gradio as gr, spaces
 from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
+from datasets import load_dataset
 
 MODEL_ID = os.getenv("MODEL_ID", "JDhruv14/Qwen2.5-3B-Gita-FT")
 
@@ -13,6 +14,7 @@ model = AutoModelForCausalLM.from_pretrained(
     trust_remote_code=True,
 )
 
+dataset = load_dataset("JDhruv14/Bhagavad-Gita-QA")
 # Ensure pad token exists (many chat models reuse EOS as PAD)
 if tokenizer.pad_token_id is None and tokenizer.eos_token_id is not None:
     tokenizer.pad_token = tokenizer.eos_token