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