Hugging Face's logo

Spaces:
banao-tech
/
OmniPar
Sleeping

App Files Community
banao-tech commited on
Commit
99b81a6
·
verified ·
1 Parent(s): 06a056f

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -23,7 +23,7 @@ from utils import (
23
  )
24
  import torch
25
 
26
- yolo_model = get_yolo_model(model_path='NewParser/best.pt')
27
  #caption_model_processor = get_caption_model_processor(model_name="florence2", model_name_or_path="icon_caption_florence")
28
 
29
  from ultralytics import YOLO
 
23
  )
24
  import torch
25
 
26
+ #yolo_model = get_yolo_model(model_path='NewParser/best.pt')
27
  #caption_model_processor = get_caption_model_processor(model_name="florence2", model_name_or_path="icon_caption_florence")
28
 
29
  from ultralytics import YOLO