Spaces:

nikigoli
/

countgd

Running on T4

nikigoli commited on Jul 7

Commit

8af2551

•

1 Parent(s): 5073050

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -166,7 +166,7 @@ args = parser.parse_args()
 #    args.device = torch.device('cuda')
 #else:
 #    args.device = torch.device('cpu')
-args.device = torch.device('cuda')
 model, transform = build_model_and_transforms(args)
 examples = [

 #    args.device = torch.device('cuda')
 #else:
 #    args.device = torch.device('cpu')
+args.device = torch.device('cpu')
 model, transform = build_model_and_transforms(args)
 examples = [