Update app.py

app.py

@@ -166,27 +166,27 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # 🔴 CHANGE THIS: your actual model repo id
 MODEL_REPO = "harishwar017/transliteration_transformer"
 
-# Download files from HF Hub into the Space’s local cache
-src_json_path = hf_hub_download(repo_id=MODEL_REPO, filename="src_stoi.json")
-tgt_json_path = hf_hub_download(repo_id=MODEL_REPO, filename="tgt_stoi.json")
+# # Download files from HF Hub into the Space’s local cache
+# src_json_path = hf_hub_download(repo_id=MODEL_REPO, filename="src_stoi.json")
+# tgt_json_path = hf_hub_download(repo_id=MODEL_REPO, filename="tgt_stoi.json")
 model_path    = hf_hub_download(repo_id=MODEL_REPO, filename="best_hindi_roman_transformer.pt")
 
-# Load vocabularies
-with open(src_json_path, "r", encoding="utf-8") as f:
-    src_stoi = json.load(f)
-
-with open(tgt_json_path, "r", encoding="utf-8") as f:
-    tgt_itos = json.load(f)
-    
 # # Load vocabularies
-# with open("src_stoi.json", "r", encoding="utf-8") as f:
+# with open(src_json_path, "r", encoding="utf-8") as f:
 #     src_stoi = json.load(f)
 
-# with open("tgt_stoi.json", "r", encoding="utf-8") as f:
-#     tgt_stoi = json.load(f)
+# with open(tgt_json_path, "r", encoding="utf-8") as f:
+#     tgt_itos = json.load(f)
+    
+# Load vocabularies
+with open("src_stoi.json", "r", encoding="utf-8") as f:
+    src_stoi = json.load(f)
+
+with open("tgt_stoi.json", "r", encoding="utf-8") as f:
+    tgt_stoi = json.load(f)
 
 # Build inverse mapping for target
-# tgt_itos = {int(v): k for k, v in tgt_stoi.items()}  # keys might be strings in JSON
+tgt_itos = {int(v): k for k, v in tgt_stoi.items()}  # keys might be strings in JSON
 
 PAD_TOKEN = "<pad>"
 SOS_TOKEN = "<sos>"