Spaces:

Aumkeshchy2003
/

Object_Detection

Sleeping

Aumkeshchy2003 commited on Feb 22

Commit

1832ba3

verified ·

1 Parent(s): 8513c99

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ from PIL import Image
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True).to(device)
 if device.type == 'cuda':
     model.half()

 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+model = torch.hub.load('ultralytics/yolov5', 'yolov5x', pretrained=True).to(device)
 if device.type == 'cuda':
     model.half()