Updated model

app.py

@@ -9,7 +9,7 @@ from resnet import ResNet18
 import gradio as gr
 
 model = ResNet18()
-model.load_state_dict(torch.load("model.ckpt", map_location=torch.device('cpu')), strict=False)
+model.load_state_dict(torch.load("model.pth", map_location=torch.device('cpu')), strict=False)
 
 inv_normalize = transforms.Normalize(
     mean=[-0.50/0.23, -0.50/0.23, -0.50/0.23],

model.ckpt → model.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:940f21f828787740b7b275a45b29051806977b52570b4e2afbb50a3f1dd04cab
-size 89492032
+oid sha256:5a7e95c31e09067216da79d276b0496bca3db976f4ec44fdae4a0b9db4b38fed
+size 44766018