adding model check

Virgula0 · Jan 18, 2025 · dbcdbd3 · dbcdbd3
1 parent 0536d5b
commit dbcdbd3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/detector.py b/detector.py
@@ -66,6 +66,11 @@ def main():
             sys.exit(-1)
 
     model = joblib.load(MODEL_PATH)
+
+    if not model:
+        console.print("[bold red]Unable to load the model[/bold red]")
+        sys.exit(-1)
+
     console.print(f"[bold green][INFO] Model loaded successfully! {type(model)}[/bold green]")
 
     if os.path.exists(RUNTIME_CAPTURE):