Spaces:
Sleeping
Sleeping
Update app.py
Browse filesFixed the correct model name to load
app.py
CHANGED
@@ -35,7 +35,7 @@ mis_classified_df = mis_classified_df.sample(frac=1)
|
|
35 |
device = torch.device("cuda")
|
36 |
|
37 |
model1 = custResNet()
|
38 |
-
model1.load_state_dict(torch.load('
|
39 |
model1.eval()
|
40 |
|
41 |
transform = transforms.Compose([
|
|
|
35 |
device = torch.device("cuda")
|
36 |
|
37 |
model1 = custResNet()
|
38 |
+
model1.load_state_dict(torch.load('custom_resnet_model.pth', map_location=torch.device('cpu')), strict=False)
|
39 |
model1.eval()
|
40 |
|
41 |
transform = transforms.Compose([
|