Spaces:
Sleeping
Sleeping
Update app.py
Browse filesCorrected the resnet model state file name
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('cust_resnet_model.pth', map_location=torch.device('cpu')), strict=False)
|
| 39 |
model1.eval()
|
| 40 |
|
| 41 |
transform = transforms.Compose([
|