Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ import gradio as gr
|
|
| 7 |
import matplotlib.pyplot as plt
|
| 8 |
|
| 9 |
# 1. 加载已训练好的分割模型
|
| 10 |
-
model = YOLO("
|
| 11 |
|
| 12 |
def analyze_video(video_path, num_mice, window_size_sec=1, fps=30):
|
| 13 |
"""
|
|
|
|
| 7 |
import matplotlib.pyplot as plt
|
| 8 |
|
| 9 |
# 1. 加载已训练好的分割模型
|
| 10 |
+
model = YOLO("fst-v1.2-n.onnx") # 模型文件需手动上传至 Space :contentReference[oaicite:3]{index=3}
|
| 11 |
|
| 12 |
def analyze_video(video_path, num_mice, window_size_sec=1, fps=30):
|
| 13 |
"""
|