Update PE_main.py

MMJohar-IT20083496 · Nov 2, 2023 · 6b2b2ae · 6b2b2ae
1 parent ccd9ec6
commit 6b2b2ae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/PE_main.py b/PE_main.py
@@ -197,6 +197,6 @@ def extract_infos(fpath):
 
     #prediciting if the PE is malicious or not based on the extracted features
     res= clf.predict([pe_features])[0]
-    print ('The file %s is %s' % (os.path.basename(sys.argv[0]),['malicious', 'legitimate'][res]))
+    print ('The file %s is %s' % (os.path.basename(sys.argv[1]),['malicious', 'legitimate'][res]))
 
     sys.exit(res)