Spaces:

GIGAParviz
/

Captcha_Time

Sleeping

GIGAParviz commited on Aug 16, 2024

Commit

c2ae097

verified ·

1 Parent(s): 95d3f9a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -85,7 +85,7 @@ path = "Captcha(Best).pt"
 from torchvision import transforms
 model = ModelWithAttention(10).to(device)
-model.load_state_dict(torch.load(path) , map_location=torch.device('cpu'))
 transform = transforms.Compose([

 from torchvision import transforms
 model = ModelWithAttention(10).to(device)
+model.load_state_dict(torch.load(path , map_location=torch.device('cpu')))
 transform = transforms.Compose([