Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 = "
|
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)
|