Update app.py

app.py

@@ -42,8 +42,14 @@ def train_and_deploy(write_token, repo_name, license_text):
     print(f"Dataset keys: {dataset.keys()}")
     if "train" not in dataset:
         raise KeyError("The dataset does not contain a 'train' split.")
+    
+    # testセットが存在しない場合、trainセットを分割してtestセットを作成
     if "test" not in dataset:
-        raise KeyError("The dataset does not contain a 'test' split.")
+        dataset = dataset["train"].train_test_split(test_size=0.1)
+        dataset = DatasetDict({
+            "train": dataset["train"],
+            "test": dataset["test"]
+        })
     
     # データセットの最初のエントリのキーを確認
     print(f"Sample keys in 'train' split: {dataset['train'][0].keys()}")
@@ -139,9 +145,9 @@ with gr.Blocks() as demo:
 
     def update_ui():
         global progress_info
-        status.update(value=progress_info["status"])
+        status.value = progress_info["status"]
         progress.update(value=progress_info["progress"])
-        time_remaining.update(value=f"{progress_info['time_remaining']}秒" if progress_info['time_remaining'] else "待機中")
+        time_remaining.value = f"{progress_info['time_remaining']}秒" if progress_info['time_remaining'] else "待機中"
 
     train_button.click(fn=train_and_deploy, inputs=[token_input, repo_input, license_input], outputs=output).then(fn=update_ui)