Spaces:

theschoolofai
/

demo

Runtime error

theschoolofai commited on Jul 28, 2023

Commit

2ac5929

1 Parent(s): 6e57808

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

 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()