Spaces:

promptsai
/

crowdcontrol

Runtime error

promptsai commited on Feb 1, 2024

Commit

c36ec65

verified ·

1 Parent(s): 7704533

Update code/demo.py

Files changed (1) hide show

code/demo.py CHANGED Viewed

@@ -12,7 +12,7 @@ model_path = "pretrained_models/model_qnrf.pth"
 url = "https://drive.google.com/uc?id=1nnIHPaV9RGqK8JHL645zmRvkNrahD9ru"
 gdown.download(url, model_path, quiet=False)
-device = torch.device('cuda')  # device can be "cpu" or "gpu"
 model = vgg19()
 model.to(device)

 url = "https://drive.google.com/uc?id=1nnIHPaV9RGqK8JHL645zmRvkNrahD9ru"
 gdown.download(url, model_path, quiet=False)
+device = torch.device('cpu')  # device can be "cpu" or "gpu"
 model = vgg19()
 model.to(device)