Steven10429 commited on
Commit
0439ea7
·
verified ·
1 Parent(s): bb4bbf2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -199,6 +199,8 @@ def process_model(base_model, lora_model, repo_name, hf_token, progress=gr.Progr
199
  login(hf_token)
200
  os.environ["HF_TOKEN"] = hf_token
201
  api = HfApi(token=hf_token)
 
 
202
 
203
  # 清空之前的日志
204
  current_logs.clear()
@@ -275,7 +277,8 @@ def create_ui():
275
  )
276
  repo_name = gr.Textbox(
277
  label="HuggingFace仓库名称",
278
- placeholder="输入要创建的仓库名称"
 
279
  )
280
  hf_token = gr.Textbox(
281
  label="HuggingFace Token",
 
199
  login(hf_token)
200
  os.environ["HF_TOKEN"] = hf_token
201
  api = HfApi(token=hf_token)
202
+ if repo_name == "Auto":
203
+ repo_name = base_model.split("/")[-1] + "_" + lora_model.split("/")[-1]
204
 
205
  # 清空之前的日志
206
  current_logs.clear()
 
277
  )
278
  repo_name = gr.Textbox(
279
  label="HuggingFace仓库名称",
280
+ placeholder="输入要创建的仓库名称",
281
+ value="Auto"
282
  )
283
  hf_token = gr.Textbox(
284
  label="HuggingFace Token",