Update app.py

app.py

@@ -183,7 +183,7 @@ def load_model_and_data():
     print(f"Using device: {device}")
 
     # Load the dataset (for vocabulary)
-    file_path = 'path_to_your_dataset.txt'  # Replace with the path to your dataset file
+    file_path = 'newcode15M.txt'  # Replace with the path to your dataset file
     dataset = TranslationDataset(file_path)
 
     # Model hyperparameters (make sure these match your trained model)
@@ -203,7 +203,7 @@ def load_model_and_data():
     ).to(device)
 
     # Load the trained model
-    model.load_state_dict(torch.load('path_to_your_model.pth', map_location=device))
+    model.load_state_dict(torch.load('AllOneLM.pth', map_location=device))
     model.eval()
 
     return model, dataset.src_vocab, dataset.tgt_vocab, device