Update app.py

app.py

@@ -1,3 +1,16 @@
+import os
+import urllib.request
+
+# Download dataset only once (if not already downloaded)
+dataset_url = "https://huggingface.co/datasets/bitext/Bitext-customer-support-llm-chatbot-training-dataset/resolve/main/Bitext_Sample_Customer_Support_Training_Dataset_27K__Labeled.csv"
+dataset_file = "bitext_dataset.csv"
+
+if not os.path.exists(dataset_file):
+    print("Downloading Bitext dataset...")
+    urllib.request.urlretrieve(dataset_url, dataset_file)
+    print("Download complete.")
+
+
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 import torch