Update app.py

app.py

@@ -232,7 +232,7 @@ class RLAgent:
         self.epsilon = max(self.epsilon_min, self.epsilon * self.epsilon_decay)
 
 
-def load_dataset():
+def load_hf_dataset():
     dataset = load_dataset(HF_DATASET, split=HF_SPLIT)
 
     data = []
@@ -405,7 +405,7 @@ def load_model():
     encoder.eval()
 
     print("Loading dataset...")
-    data = load_dataset()
+    data = load_hf_dataset()
     print(f"Dataset size: {len(data)} examples")
 
     print("Training RL agent...")