ZiqianLiu commited on
Commit
4b1db35
1 Parent(s): 7de540a

Upload nodes.py

Browse files
Files changed (1) hide show
  1. nodes.py +2 -1
nodes.py CHANGED
@@ -365,7 +365,8 @@ from ultralytics import YOLO
365
  from PIL import Image
366
 
367
  # Load a pretrained YOLOv8n model
368
- model = YOLO(task='detect', model='/root/autodl-tmp/ComfyUI/custom_nodes/comfyui-reactor-node/yolov8m_200e.pt')
 
369
 
370
  class Mynode_2:
371
  @classmethod
 
365
  from PIL import Image
366
 
367
  # Load a pretrained YOLOv8n model
368
+ current_directory = os.getcwd()
369
+ model = YOLO(task='detect', model=current_directory + '/custom_nodes/yolov8_face/yolov8m_200e.pt')
370
 
371
  class Mynode_2:
372
  @classmethod