Update app.py

app.py

@@ -136,11 +136,11 @@ class Transformer(nn.Module):
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # Load tokenizers
-sp_pseudo = spm.SentencePieceProcessor(model_file="pseudo.model")  # For decoding pseudocode (target)
-sp_code = spm.SentencePieceProcessor(model_file="code.model")      # For encoding C++ (source)
+sp_pseudo = spm.SentencePieceProcessor(model_file="pseudocode_tokenizer.model")  # For decoding pseudocode (target)
+sp_code = spm.SentencePieceProcessor(model_file="code_tokenizer.model")      # For encoding C++ (source)
 
 # Load the full saved model (architecture + weights)
-model_path = "transformer_cpp_to_pseudo.pth"
+model_path = "code2pseudo.pth"
 model = torch.load(model_path, map_location=device, weights_only=False)
 model.eval()
 model = model.to(device)