Spaces:

stanimirovb
/

lenet-digit-classifier

Sleeping

stanimirovb commited on Dec 8, 2023

Commit

fae33e7

verified ·

1 Parent(s): 2f12319

how about this?

Files changed (1) hide show

app.py CHANGED Viewed

@@ -35,7 +35,7 @@ class LeNet(nn.Module):
         return nn.functional.softmax(out[0], dim = 0)
 lenet = LeNet()
-lenet.load_state_dict(torch.load('ibob-lenet-v1/lenet-v1.pth', map_location='cpu'))
 resize = torchvision.transforms.Resize((28, 28), antialias=True)

         return nn.functional.softmax(out[0], dim = 0)
 lenet = LeNet()
+lenet.load_state_dict(torch.load('stanimirovb/ibob-lenet-v1/lenet-v1.pth', map_location='cpu'))
 resize = torchvision.transforms.Resize((28, 28), antialias=True)