Spaces:

Catmeow
/

Count_objects_in_picture

Runtime error

Catmeow commited on Nov 3, 2022

Commit

db223bc

1 Parent(s): 7b330ff

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import torch
 from PIL import Image
 import pandas as pd
-model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True).cuda()
 def inference(im):
     results = model(im)

 from PIL import Image
 import pandas as pd
+model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
 def inference(im):
     results = model(im)