anonymousatom commited on
Commit
79c9d6f
·
verified ·
1 Parent(s): dd25c55

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ from ultralytics import YOLO
2
+
3
+ model = YOLO('best.pt')
4
+
5
+ image_path = input("Enter image path: ")
6
+ results = model(image_path, device = 0, amp = 'True') # return a list of Results objects
7
+
8
+ print(results)