Update app.py

app.py

@@ -15,11 +15,12 @@ st.write("castorini/afriberta-corpusを使って、ユーザーが入力した
 model_name = st.text_input("トレーニングするモデル名 (例: Qwen/Qwen2.5-1.5B-Instruct)")
 dataset_name = "castorini/afriberta-corpus"
 hf_token = st.text_input("Hugging Face Write トークン", type="password")
+repo_name = st.text_input("Hugging Faceリポジトリ名")  # ユーザーが入力できるリポジトリ名
 output_dir = "./finetuned_model"
 
 if st.button("トレーニング開始"):
-    if not model_name or not hf_token:
-        st.warning("モデル名とトークンを入力してください")
+    if not model_name or not hf_token or not repo_name:
+        st.warning("モデル名、トークン、リポジトリ名を入力してください")
     else:
         # トークンの設定
         HfFolder.save_token(hf_token)
@@ -68,7 +69,6 @@ if st.button("トレーニング開始"):
         trainer.save_model(output_dir)
 
         # Hugging Face Hub にデプロイ
-        repo_name = model_name.split('/')[-1] + "-finetuned"
         api = HfApi()
         api.create_repo(repo_name, token=hf_token)
         repo = Repository(local_dir=output_dir, clone_from=repo_name, use_auth_token=hf_token)