Spaces:

theschoolofai
/

demo

Runtime error

theschoolofai commited on Jul 28, 2023

Commit

bdd18b5

1 Parent(s): 2ac5929

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ from resnet import ResNet18
 import gradio as gr
 model = ResNet18()
-model.load_state_dict(torch.load("model.pth"), strict=False, map_location=torch.device('cpu'))
 def inference(input_img, transparency):
     transform = transforms.ToTensor()

 import gradio as gr
 model = ResNet18()
+model.load_state_dict(torch.load("model.pth", map_location=torch.device('cpu')), strict=False)
 def inference(input_img, transparency):
     transform = transforms.ToTensor()