Spaces:
Sleeping
Sleeping
AdarshRajDS commited on
Commit ·
6d5d66b
1
Parent(s): bef3d34
Fix Resnet with model
Browse files
app.py
CHANGED
|
@@ -29,7 +29,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 29 |
# Load baseline model (ResNet)
|
| 30 |
# ------------------
|
| 31 |
resnet_ckpt = torch.load(
|
| 32 |
-
"
|
| 33 |
map_location=device
|
| 34 |
)
|
| 35 |
|
|
|
|
| 29 |
# Load baseline model (ResNet)
|
| 30 |
# ------------------
|
| 31 |
resnet_ckpt = torch.load(
|
| 32 |
+
"resnet50_multitask_mold.pth",
|
| 33 |
map_location=device
|
| 34 |
)
|
| 35 |
|