tree3po commited on
Commit
80826f2
·
verified ·
1 Parent(s): 2b85a9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -11
app.py CHANGED
@@ -9,19 +9,12 @@ import numpy as np
9
  import uuid
10
  import spaces
11
 
12
- print(YOLO)
13
-
14
- """| YOLOv8 | `yolov8n.pt` `yolov8s.pt` `yolov8m.pt` `yolov8l.pt` `yolov8x.pt` …
15
- | YOLOv8-seg | `yolov8n-seg.pt` `yolov8s-seg.pt` `yolov8m-seg.pt` `yolov8l-seg.pt` `yolov8x-seg.pt` …
16
- | YOLOv8-pose | `yolov8n-pose.pt` `yolov8s-pose.pt` `yolov8m-pose.pt` `yolov8l-pose.pt` `yolov8x-pose.pt` `yolov8x-pose…
17
- | YOLOv8-obb | `yolov8n-obb.pt` `yolov8s-obb.pt` `yolov8m-obb.pt` `yolov8l-obb.pt` `yolov8x-obb.pt` …
18
- | YOLOv8-cls | `yolov8n-cls.pt` `yolov8s-cls.pt` `yolov8m-cls.pt` `yolov8l-cls.pt` `yolov8x-cls.pt` """
19
-
20
- ver=[6:11]
21
  ltr=["n","s","m","1","x"]
22
- tsk=["","seg","pose","obb","cls"]
23
  #yolov8s.pt
24
- modin=f"yolov{ver[2]}{ltr[1]}-{tsk[0]}.pt"
 
25
  print(modin)
26
  model = YOLO(modin)
27
 
 
9
  import uuid
10
  import spaces
11
 
12
+ ver=[6,7,8.9,10,11]
 
 
 
 
 
 
 
 
13
  ltr=["n","s","m","1","x"]
14
+ tsk=["","-seg","-pose","-obb","-cls"]
15
  #yolov8s.pt
16
+ modin=f"yolov{ver[2]}{ltr[1]}{tsk[0]}.pt"
17
+ if
18
  print(modin)
19
  model = YOLO(modin)
20