updated download script and requirements

download_model.py

@@ -1,7 +1,10 @@
 import os
-import gdown
+from huggingface_hub import hf_hub_download
 
-modelsave_name = 'model_parameters.pt'
-if os.path.isfile(modelsave_name) == False:
-    url = 'https://drive.google.com/file/d/1_mYn2LrhG080Xvt26tWBtJ8U_0F2E1-s/view?usp=sharing'
-    gdown.download(url, output=modelsave_name, fuzzy=True)
+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='./'
+    )

requirements.txt

@@ -10,6 +10,7 @@ fsspec==2024.5.0
 gdown==5.2.0
 h11==0.14.0
 htmltools==0.6.0
+huggingface-hub==0.28.1
 idna==3.7
 Jinja2==3.1.4
 linkify-it-py==2.0.3
@@ -27,6 +28,7 @@ pillow==10.3.0
 prompt_toolkit==3.0.50
 PySocks==1.7.1
 python-multipart==0.0.20
+PyYAML==6.0.2
 questionary==2.1.0
 requests==2.31.0
 setuptools==75.8.0