Mtemel commited on
Commit
7ebb272
·
verified ·
1 Parent(s): 135e5c6

Main app crt

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ from ultralytics import YOLOvv8
2
+
3
+ model = YOLOvv8.from_pretrained("keremberke/yolov8n-pcb-defect-segmentation")
4
+ source = 'http://images.cocodataset.org/val2017/000000039769.jpg'
5
+ model.predict(source=source, save=True)