File size: 1,352 Bytes
e9f9fd3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
import hashlib
import requests
import os
def get_file_hash(filepath):
sha256_hash = hashlib.sha256()
with open(filepath, "rb") as f:
for byte_block in iter(lambda: f.read(4096), b""):
sha256_hash.update(byte_block)
return sha256_hash.hexdigest()
def download_file(url, filename):
print(f"Downloading {url}...")
r = requests.get(url, stream=True)
with open(filename, 'wb') as f:
for chunk in r.iter_content(chunk_size=8192):
f.write(chunk)
return filename
external_urls = {
"glitch_art.onnx": "https://huggingface.co/thookham/DeOldify-on-Browser/resolve/main/deoldify-art.onnx",
"glitch_quant.onnx": "https://huggingface.co/thookham/DeOldify-on-Browser/resolve/main/deoldify-quant.onnx"
}
# Local files
local_files = {
"local_art.onnx": "models/deoldify-art.onnx",
"local_quant.onnx": "models/deoldify-quant.onnx"
}
print("--- Local File Hashes ---")
for name, path in local_files.items():
if os.path.exists(path):
print(f"{name}: {get_file_hash(path)}")
else:
print(f"{name}: File not found")
print("\n--- External File Hashes ---")
for name, url in external_urls.items():
try:
download_file(url, name)
print(f"{name}: {get_file_hash(name)}")
except Exception as e:
print(f"Error downloading {name}: {e}")
|