neko941 commited on
Commit
3710aea
1 Parent(s): c9ce858

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -45,6 +45,7 @@ if image != '':
45
  image = check_file(image)
46
  input_image = Image.open(image)
47
  model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')
48
- results = model(input_image, size=640)
49
  for img in results.render():
50
  st.image(img)
 
 
45
  image = check_file(image)
46
  input_image = Image.open(image)
47
  model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')
48
+ results = model(input_image, size=1280)
49
  for img in results.render():
50
  st.image(img)
51
+ st.write(results.pandas().xyxy[0])