Update app.py

app.py

@@ -33,11 +33,11 @@ if st.button("トレーニング開始"):
         # データセットの準備（スワヒリ語）
         st.write("データセットのロード中...")
         from datasets import load_dataset
-        dataset = load_dataset(dataset_name, 'swahili', split="train")  # 言語を指定
+        dataset = load_dataset(dataset_name, split="train")  # 言語を指定
 
         # トレーニング用のデータセットの準備
         def tokenize_function(examples):
-            return tokenizer(examples["text"], padding="max_length", truncation=True, max_length=128)
+            return tokenizer(examples["text"], padding="max_length", truncation=True, max_length=168)
 
         tokenized_dataset = dataset.map(tokenize_function, batched=True)