Hakureirm commited on
Commit
8b79e99
·
verified ·
1 Parent(s): cfc7448

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,7 @@ APP_PASSWORD = os.getenv("APP_PASSWORD", "default_password") # 从环境变量
16
 
17
  app = FastAPI()
18
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
19
- model = YOLO('kunin-mice-pose.v0.1.3.pt')
20
 
21
  # 定义认证状态
22
  class AuthState:
@@ -85,7 +85,7 @@ def process_video(video_path, process_seconds=20, conf_threshold=0.2, max_det=8)
85
  vid_stride=1,
86
  max_det=max_det,
87
  retina_masks=True, # 更精细的显示
88
- verbose=True
89
  )
90
 
91
  # 处理结果
 
16
 
17
  app = FastAPI()
18
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
19
+ model = YOLO('kunin-mice-pose.v0.1.4pt')
20
 
21
  # 定义认证状态
22
  class AuthState:
 
85
  vid_stride=1,
86
  max_det=max_det,
87
  retina_masks=True, # 更精细的显示
88
+ verbose=False
89
  )
90
 
91
  # 处理结果