Ryonaly commited on
Commit
8c8d8de
·
verified ·
1 Parent(s): b161bf6

Create pipeline/medivlad_wrapper.py

Browse files
Files changed (1) hide show
  1. pipeline/medivlad_wrapper.py +8 -0
pipeline/medivlad_wrapper.py ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ def is_ready():
3
+ return os.path.isdir("external/MeDiVLAD"), os.path.isdir("models/medivlad")
4
+ def run_video_severity(_clip):
5
+ repo_ok, weights_ok = is_ready()
6
+ if not repo_ok: raise RuntimeError("Original repo missing: external/MeDiVLAD")
7
+ if not weights_ok: return {"status":"weights-missing","message":"Place MeDiVLAD weights in models/medivlad/"}
8
+ return {"severity_score": None, "status": "not-implemented-yet"}