abdullah63 commited on
Commit
2fc123c
·
verified ·
1 Parent(s): 32ff591

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ sp_code = spm.SentencePieceProcessor(model_file="code.model")
12
 
13
  # Load the full saved model (architecture + weights)
14
  model_path = "pseudo-to-cpp-model.pth" # Adjust path as needed
15
- model = torch.load(model_path, map_location=device)
16
  model.eval()
17
  model = model.to(device)
18
 
 
12
 
13
  # Load the full saved model (architecture + weights)
14
  model_path = "pseudo-to-cpp-model.pth" # Adjust path as needed
15
+ model = torch.load(model_path, map_location=device, weights_only=False)
16
  model.eval()
17
  model = model.to(device)
18