dataset 選択

run.py

@@ -57,17 +57,17 @@ def toggle_buttons(profile: gr.OAuthProfile | None):
     return gr.update(visible=False), gr.update(visible=True)
 
 
-def load_data(profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken | None) -> str:
+def load_data(choice_dataset, profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken | None) -> str:
     if oauth_token is None:
-        return "Please log in to load your datasets."
-    dataset = load_dataset("kanhatakeyama/OrcaJaMixtral8x22b")
+        return "ログインしてからデータセットを表示してください。"
+    dataset = load_dataset(choice_dataset)
     return dataset
 
 
-def display_dataset(profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken | None):
+def display_dataset(choice_dataset, profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken | None):
     if profile is None:
         return gr.update(visible=True, value="ログインしてデータセットを表示してください。"), None, None
-    dataset = load_data(profile, oauth_token=oauth_token.token)
+    dataset = load_data(choice_dataset, profile, oauth_token)
     if dataset:
         # "train" スプリットの最初のデータを取得
         first_data = dataset['train'][0]
@@ -90,7 +90,7 @@ theme_ = gr.State("Light")
 
 with gr.Blocks(theme=theme_, css=load_css()) as demo:
 
-    gr.Markdown("# 自動生成データセット アノテーション for Tanuki 2Phase")
+    gr.Markdown("# データセット アノテーション for Tanuki (2Phase)")
 
     with gr.Tab("アノテーション"):
 
@@ -118,7 +118,8 @@ with gr.Blocks(theme=theme_, css=load_css()) as demo:
 
         with gr.Row():
 
-            choice_dataset = gr.Dropdown(label="アノテーションするデータセット", choices=dataset_list,
+            choice_dataset = gr.Dropdown(label="アノテーションするデータセット",
+                                        choices=dataset_list,
                                         value="a", elem_id="dataset_sel")
 
             data_load_btn = gr.Button("データセットを読み込む")
@@ -135,15 +136,14 @@ with gr.Blocks(theme=theme_, css=load_css()) as demo:
         answer_text = gr.Textbox(label="回答:　改行はShift+Enterです。", elem_id="answer", interactive=True)
 
         data_load_btn.click(
-            display_dataset,
-            inputs=None,
-            outputs=[dataset_display, question_text, answer_text],
-        )
+                display_dataset,
+                inputs=[choice_dataset],
+                outputs=[dataset_display, question_text, answer_text],
+            )
 
         def on_submit(answer_text):
             return f" {answer_text}"
 
-
     with gr.Tab("アノテ済みデータセット"):
 
         gr.Textbox("データセットID", lines=1, placeholder="データセットIDを入力してください。")