Update app.py

app.py

@@ -52,17 +52,9 @@ class Vocab:
 
 
 # ------------- load vocab from cache -------------
-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"]
-vocab       = cache["vocab"]
-
-# safety: rebuild idx2word if needed
-if not hasattr(vocab, "idx2word") or len(vocab.idx2word) != len(vocab.word2idx):
-    vocab.idx2word = {i: w for w, i in vocab.word2idx.items()}
+print("Loading vocab...")
+data = torch.load("ubuntu_vocab_only.pt", map_location="cpu", weights_only=False)
+vocab = data["vocab"]
 
 PAD_IDX = vocab.word2idx["<PAD>"]
 SOS_IDX = vocab.word2idx["<SOS>"]