Upload app.py

app.py

@@ -6,6 +6,8 @@ import logging
 
 HF_TOKEN = os.environ["HF_TOKEN"] if "HF_TOKEN" in os.environ else ""
 WiseModel_TOKEN = os.environ["WM_TOKEN"] if "WM_TOKEN" in os.environ else ""
+GIT_USER = os.environ["GIT_USE"] if "GIT_USER" in os.environ else ""
+GIT_EMAIL = os.environ["GIT_EMAIL"] if "GIT_EMAIL" in os.environ else ""
 
 
 def get_cache_dir():
@@ -21,6 +23,15 @@ def check_disk():
     return os.system("df -h /")
 
 
+def init_git_user(username: str, email: str):
+
+    if not username or not email:
+        raise gr.Error("Please config your git username and email in the environment variables.")
+
+    os.system(f"git config --global user.email {email}")
+    os.system(f"git config --global user.name {username}")
+
+
 def pull_from_wisemodel(
     token: str, url: str, repo_name: str, cache_dir, branch: Optional[str] = None
 ):
@@ -147,6 +158,7 @@ def handle(wisemodel_link, hf_repo_id):
         # (check_disk, (), {}),
         # (push_to_hf, (cache_dir, wisemodel_repo_name, hf_repo_id), {}),
         # (check_disk, (), {}),
+        (init_git_user, (GIT_USER, GIT_EMAIL), {}),
         (pull_from_hf, (cache_dir, hf_repo_id), {}),
         (remove_file, (hf_repo_id, cache_dir), {}),
         (check_disk, (), {}),
@@ -198,8 +210,8 @@ with gr.Blocks() as demo:
 # 如何使用这个空间？
 # How to use this Space?
 
-- 点击右上角settings后面的“…”按钮，选择“Duplicate this Space”创建一个私有的space，同时输入wisemodel的token(必填)，以及Huggingface的token（必填），确保有相应repo写入的权限。
-- Duplicate this Space and providing WiseModel token (mandatory) and your read/write HF token (mandatory).
+- 点击右上角settings后面的“…”按钮，选择“Duplicate this Space”创建一个私有的space，同时输入wisemodel的token(必填)，以及Huggingface的token（必填），确保有相应repo写入的权限。你还需要输入你的git用户名和邮箱。
+- Duplicate this Space and providing WiseModel token (mandatory) and your read/write HF token (mandatory).you also need to provide your git username and email to push the model to WiseModel.
 
 - 在wiseModel上创建一个空的repo，这一步需要手动完成，Space不会为您创建一个空的repo。
 - Create your target model repo on WiseModel. This step is not automated.