Update app.py

app.py

@@ -56,8 +56,8 @@ if not os.path.exists(CACHE_FILE):
     raise FileNotFoundError(f"Cache file {CACHE_FILE} not found. Please upload ubuntu_data_cache.pt")
 
 cache = torch.load(CACHE_FILE, map_location="cpu", weights_only=False)
-train_pairs = cache["train"]
-test_pairs  = cache["test"]
+# train_pairs = cache["train"]
+# test_pairs  = cache["test"]
 vocab       = cache["vocab"]
 
 # safety: rebuild idx2word if needed