Update pipeline/medivlad_wrapper.py

pipeline/medivlad_wrapper.py

@@ -1,8 +1,14 @@
 import os
+
 def is_ready():
-    return os.path.isdir("external/MeDiVLAD"), os.path.isdir("models/medivlad")
+    repo_ok = os.path.isdir("external/MeDiVLAD")
+    weights_ok = os.path.isdir("models/medivlad")
+    return repo_ok, weights_ok
+
 def run_video_severity(_clip):
     repo_ok, weights_ok = is_ready()
-    if not repo_ok: raise RuntimeError("Original repo missing: external/MeDiVLAD")
-    if not weights_ok: return {"status":"weights-missing","message":"Place MeDiVLAD weights in models/medivlad/"}
+    if not repo_ok:
+        raise RuntimeError("Original repo missing: external/MeDiVLAD")
+    if not weights_ok:
+        return {"status": "weights-missing", "message": "Place MeDiVLAD weights in models/medivlad/"}
     return {"severity_score": None, "status": "not-implemented-yet"}