girish00 commited on
Commit
de2ac96
·
verified ·
1 Parent(s): 008e861

add cloud structured inference wrapper

Browse files
Files changed (1) hide show
  1. upload_to_hf.py +29 -0
upload_to_hf.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import argparse
2
+ import os
3
+ from huggingface_hub import upload_folder
4
+
5
+
6
+ def main():
7
+ parser = argparse.ArgumentParser()
8
+ parser.add_argument("--force", action="store_true")
9
+ parser.add_argument("--model-dir", type=str, default="./model")
10
+ parser.add_argument("--repo-id", type=str, required=True)
11
+ args = parser.parse_args()
12
+
13
+ if not os.path.exists(args.model_dir):
14
+ raise FileNotFoundError(
15
+ f"Model directory not found: {args.model_dir}. Train first before upload."
16
+ )
17
+
18
+ upload_folder(
19
+ folder_path=args.model_dir,
20
+ repo_id=args.repo_id,
21
+ repo_type="model",
22
+ commit_message="update model",
23
+ ignore_patterns=[] # ensures upload always checks properly
24
+ )
25
+ print(f"Uploaded model to: {args.repo_id}")
26
+
27
+
28
+ if __name__ == "__main__":
29
+ main()