Spaces:

dusaurabh
/

erav2s13demo

Sleeping

dusaurabh commited on May 10

Commit

3fd124e

•

1 Parent(s): 9c69c76

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ from resnet import ResNet18
 import gradio as gr
 model = ResNet18()
-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],

 import gradio as gr
 model = ResNet18()
+model.load_state_dict(torch.load("cifar10_saved_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],