Spaces:

nikigoli
/

countgd

Sleeping

nikigoli commited on Jul 6

Commit

ddb9d72

•

1 Parent(s): 809b37d

Forced cpu usage

Files changed (1) hide show

app.py CHANGED Viewed

@@ -104,7 +104,7 @@ def get_args_parser():
 # Get counting model.
 def build_model_and_transforms(args):
     if torch.cuda.is_available():
-        args.device = torch.device('cuda')
     else:
         args.device = torch.device('cpu')
     normalize = T.Compose(

 # Get counting model.
 def build_model_and_transforms(args):
     if torch.cuda.is_available():
+        args.device = torch.device('cpu')
     else:
         args.device = torch.device('cpu')
     normalize = T.Compose(