burhan112 commited on
Commit
986c39d
·
verified ·
1 Parent(s): 2c0ed7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -140,7 +140,7 @@ sp_pseudo = spm.SentencePieceProcessor(model_file="pseudocode_tokenizer.model")
140
  sp_code = spm.SentencePieceProcessor(model_file="code_tokenizer.model") # For encoding C++ (source)
141
 
142
  # Load the full saved model (architecture + weights)
143
- model_path = "c2p.pth"
144
  model = torch.load(model_path, map_location=device, weights_only=False)
145
  model.eval()
146
  model = model.to(device)
 
140
  sp_code = spm.SentencePieceProcessor(model_file="code_tokenizer.model") # For encoding C++ (source)
141
 
142
  # Load the full saved model (architecture + weights)
143
+ model_path = "code2pseudo.pth"
144
  model = torch.load(model_path, map_location=device, weights_only=False)
145
  model.eval()
146
  model = model.to(device)