1

dl.py

@@ -1,11 +1,13 @@
 #!/usr/bin/env python
 
 import os
-
+import hashlib
 if not os.path.exists('./.gitattributes') or '35efec71055ba0c3b7c7a1bcf499d0922398679a' != hashlib.sha1(open('./.gitattributes', 'rb').read()).hexdigest():
     os.system('curl -L "https://huggingface.co/AI-Porn/AI-PORN/resolve/main/./.gitattributes" --create-dirs -o "././.gitattributes"')
-if not os.path.exists('./gen.py') or 'aa70041d002fc25357c8ac5d5137bbc5510b84f4' != hashlib.sha1(open('./gen.py', 'rb').read()).hexdigest():
+if not os.path.exists('./gen.py') or '9d27d063b324a6a71af0565b28c50d89f7925a57' != hashlib.sha1(open('./gen.py', 'rb').read()).hexdigest():
     os.system('curl -L "https://huggingface.co/AI-Porn/AI-PORN/resolve/main/./gen.py" --create-dirs -o "././gen.py"')
+if not os.path.exists('./dl.py') or '3a2186993fdd559e8d94fa0b1266681c2fdb20fe' != hashlib.sha1(open('./dl.py', 'rb').read()).hexdigest():
+    os.system('curl -L "https://huggingface.co/AI-Porn/AI-PORN/resolve/main/./dl.py" --create-dirs -o "././dl.py"')
 if not os.path.exists('./embeddings/corneo_pov_anal.pt') or '37577b40588e2c267cbaf13f3edd014d5fe1f24b' != hashlib.sha1(open('./embeddings/corneo_pov_anal.pt', 'rb').read()).hexdigest():
     os.system('curl -L "https://huggingface.co/AI-Porn/AI-PORN/resolve/main/./embeddings/corneo_pov_anal.pt" --create-dirs -o "././embeddings/corneo_pov_anal.pt"')
 if not os.path.exists('./embeddings/easynegative.pt') or '199b0109efbafa5b49abb413e56eb2a086e378eb' != hashlib.sha1(open('./embeddings/easynegative.pt', 'rb').read()).hexdigest():

gen.py

@@ -16,7 +16,7 @@ for root, dirs, files in os.walk("."):
 
 
 with open("dl.py", "w") as f:
-    f.write("#!/usr/bin/env python\n\nimport os\n\n")
+    f.write("#!/usr/bin/env python\n\nimport os\nimport hashlib\n")
     for file_path, sha1 in sha1s.items():
         f.write(f"if not os.path.exists('{file_path}') or '{sha1}' != hashlib.sha1(open('{file_path}', 'rb').read()).hexdigest():\n")
         f.write(f"    os.system('curl -L \"https://huggingface.co/AI-Porn/AI-PORN/resolve/main/{file_path}\" --create-dirs -o \"./{file_path}\"')\n")