Update app.py

app.py

@@ -52,25 +52,7 @@ class Vocab:
 
 
 # ------------- load vocab from cache -------------
-if not os.path.exists(CACHE_FILE):
-    raise FileNotFoundError(
-        f"{CACHE_FILE} not found in Space. Upload the same file you used locally."
-    )
-
-# cache structure in your notebook: {'data': pairs, 'vocab': vocab}
-cache = torch.load(CACHE_FILE, map_location="cpu", weights_only=False)
-
-if not isinstance(cache, dict) or "vocab" not in cache:
-    raise RuntimeError(
-        f"{CACHE_FILE} does not contain a 'vocab' key. "
-        f"Found keys: {list(cache.keys()) if isinstance(cache, dict) else type(cache)}"
-    )
-
-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()}
+vocab = torch.load(CACHE_FILE, map_location="cpu", weights_only=False)
 
 PAD_IDX = vocab.word2idx["<PAD>"]
 SOS_IDX = vocab.word2idx["<SOS>"]