From c64bcd23d313541bfa5c536177828c14ee19ecb8 Mon Sep 17 00:00:00 2001
From: cmdr2 <secondary.cmdr2@gmail.com>
Date: Wed, 30 Nov 2022 13:38:22 +0530
Subject: [PATCH] Picklescanner is mandatory

---
 ui/server.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ui/server.py b/ui/server.py
index 1313a0a0..fca1aca8 100644
--- a/ui/server.py
+++ b/ui/server.py
@@ -7,6 +7,7 @@ import traceback
 
 import sys
 import os
+import picklescan.scanner
 import rich
 
 SD_DIR = os.getcwd()
@@ -234,7 +235,6 @@ async def setAppConfig(req : SetAppConfigRequest):
 
 def is_malicious_model(file_path):
     try:
-        import picklescan.scanner
         scan_result = picklescan.scanner.scan_file_path(file_path)
         if scan_result.issues_count > 0 or scan_result.infected_files > 0:
             rich.print(":warning: [bold red]Scan %s: %d scanned, %d issue, %d infected.[/bold red]" % (file_path, scan_result.scanned_files, scan_result.issues_count, scan_result.infected_files))