Update app.py

app.py

@@ -37,9 +37,16 @@ class ChatDataset(Dataset):
 
 # Load training data from Hugging Face
 def load_training_data():
-    dataset = load_dataset("training_data.txt", split="train")  # Replace with actual dataset path
-    texts = [item["text"] for item in dataset if "text" in item]
-    return texts
+    try:
+        with open("training_data.txt", "r", encoding="utf-8") as file:
+            texts = [line.strip() for line in file if line.strip()]
+        return texts
+    except FileNotFoundError:
+        print("Error: training_data.txt not found in the root directory.")
+        return []
+    except Exception as e:
+        print(f"Error reading training_data.txt: {e}")
+        return []
 
 # Fine-tune model with differential privacy
 def train_model():