Upload /check_protec.py with huggingface_hub

check_protec.py

@@ -0,0 +1,25 @@
+import torch
+def validate_hash(file_path):
+    import hashlib
+
+    try:
+        # Load file
+        data = torch.load(file_path)
+
+        # Recalculate hash
+        metadata = data["metadata"]
+        original_hash = data["hash"]
+        recalculated_hash = hashlib.sha256(str(metadata).encode()).hexdigest()
+
+        if original_hash == recalculated_hash:
+            print("Valid metadata.")
+            return True
+        else:
+            print("Metadata tampered.")
+            return False
+    except Exception as e:
+        print(f"Error validating hash: {e}")
+        return False
+
+
+validate_hash("vae_weights.pth")