Gender-CNN-Demo / download_model.py
Harry2687's picture
updated download script and requirements
5b0d0fe
raw
history blame contribute delete
255 Bytes
import os
from huggingface_hub import hf_hub_download
model_name = 'resnetModel_128_epoch_2.pt'
if os.path.isfile(model_name) == False:
hf_hub_download(
repo_id='Harry2687/Gender-CNN',
filename=model_name,
local_dir='./'
)