diff --git a/projects/CV/ex.py b/projects/CV/ex.py
index 9a656bf..3576f9b 100644
--- a/projects/CV/ex.py
+++ b/projects/CV/ex.py
@@ -1,23 +1,27 @@
 from ultralytics import YOLO
 
-model = YOLO('yolov8m.pt')
+model = YOLO('yolov8s.pt')
 model.save_dir = './predict/'
 
 result = model.predict(
     './Assets/cat_dog.jpg',
     save=True,
-    device='cpu',
+    project='test',
+    name='ololo',
+    device=0,
     show=False,
     classes=[16, 15],
 )[0]
-result = model(
-    './Assets/cat_dog.jpg',
-    save=True,
-    project='test',
-    name='ololo',
-    device='cpu',
-    show=False,
-)[0]
+# result = model(
+#     './Assets/cat_dog.jpg',
+#     save=True,
+#     project='test',
+#     name='ololo',
+#     device='gpu',
+#     show=False,
+# )[0]
+
+print(model.names)
 
 # print(len(result.boxes))