Update hf.py: Add file_url arg

rabilrbl · Jan 4, 2024 · 8c5b158 · 8c5b158
1 parent ab31e12
commit 8c5b158
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/hf.py b/hf.py
@@ -2,8 +2,9 @@
 from huggingface_hub import HfApi,login, logout
 api = HfApi()
 
-repo = sys.argv[1]
-file = sys.argv[2]
+file_url = sys.argv[1]
+repo = sys.argv[2]
+file = sys.argv[3]
 
 token = os.environ.get("HF_TOKEN", None)
 
@@ -45,7 +46,7 @@ def extract_repo_id(file_url):
 tags:
     - llamafile
     - GGUF
-base_model: {extract_repo_id(file)}
+base_model: {extract_repo_id(file_url)}
 ---
 ## {"".join(repo.split("/")[1:])}