diff --git a/app.py b/app.py index 4822a03..9455dd7 100644 --- a/app.py +++ b/app.py @@ -68,7 +68,7 @@ async def random_url(_): def load_processor_model( processor_name: str, model_name: str ) -> Tuple[CLIPProcessor, CLIPModel]: - processor = CLIPProcessor.from_pretrained(processor_name) + processor = CLIPProcessor.from_pretrained(process_name) model = CLIPModel.from_pretrained(model_name) return processor, model diff --git a/modules/advanced-zero-click-deployment-interface/FlowSteering/ApplicationCode/EndUserCode/EndUserClient.py b/modules/advanced-zero-click-deployment-interface/FlowSteering/ApplicationCode/EndUserCode/EndUserClient.py index 31df0b4..eeca626 100644 --- a/modules/advanced-zero-click-deployment-interface/FlowSteering/ApplicationCode/EndUserCode/EndUserClient.py +++ b/modules/advanced-zero-click-deployment-interface/FlowSteering/ApplicationCode/EndUserCode/EndUserClient.py @@ -26,7 +26,7 @@ CycleNewEmails = None BaseEmails_directory = None # Define the default image to be sent in case of network errors -default_image='' +default_image='FlowSteering/assets/PerturbatedImages/DjiPerturbClassForward.png' def receive_complete_data(client_socket): # this function is used to receive the complete data from the client, adjust the parameters as needed based on your network conditions diff --git a/modules/advanced_malware_analysis.py b/modules/advanced_malware_analysis.py index 7d996a5..4cf2a61 100644 --- a/modules/advanced_malware_analysis.py +++ b/modules/advanced_malware_analysis.py @@ -2,6 +2,7 @@ import subprocess import os import json +import shlex class AdvancedMalwareAnalysis: def __init__(self): @@ -18,9 +19,9 @@ def analyze_malware(self, malware_path): def run_sandbox(self, malware_path): logging.info(f"Running malware in sandbox: {malware_path}") # Placeholder for sandbox execution logic - sandbox_command = f"{self.sandbox_path} {malware_path}" + sandbox_command = shlex.split(f"{self.sandbox_path} {malware_path}") try: - subprocess.run(sandbox_command, shell=True, check=True) + subprocess.run(sandbox_command, check=True) except subprocess.CalledProcessError as e: logging.error(f"Sandbox execution failed: {e}")