hf-model

app.py

@@ -7,6 +7,11 @@ import cv2
 from drexel_metadata.gen_metadata import gen_metadata
 from PIL import Image
 import urllib.request
+from huggingface_hub import hf_hub_download
+
+
+# Download model if not already cached locally
+hf_hub_download(repo_id="imageomics/Drexel-metadata-generator", filename="model_final.pth", local_dir="output/enhanced")
 
 
 EXAMPLE_URLS = [

output/enhanced/model_final.pth

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:4c021ad75105359524d9165c8ccec0be9de02b253eaf836c6916a6032f5c98fc
-size 176294747

requirements.txt

@@ -23,6 +23,7 @@ fvcore==0.1.5.post20220512
 google-auth==2.13.0
 google-auth-oauthlib==0.4.6
 grpcio==1.50.0
+huggingface-hub==0.16.4
 hydra-core==1.2.0
 idna==3.4
 imageio==2.22.2