Update app.py
Browse files
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",
|