Update app.py

app.py

@@ -1,5 +1,5 @@
 # 必要なライブラリをインストールしておいてください
-# pip install streamlit transformers torch huggingface_hub
+# pip install streamlit transformers torch huggingface_hub datasets
 
 import streamlit as st
 from transformers import AutoModelForCausalLM, AutoTokenizer, Trainer, TrainingArguments
@@ -30,10 +30,10 @@ if st.button("トレーニング開始"):
         model = AutoModelForCausalLM.from_pretrained(model_name)
         tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-        # データセットの準備
+        # データセットの準備（スワヒリ語）
         st.write("データセットのロード中...")
         from datasets import load_dataset
-        dataset = load_dataset(dataset_name, split="swahili")
+        dataset = load_dataset(dataset_name, 'swahili', split="train")  # 言語を指定
 
         # トレーニング用のデータセットの準備
         def tokenize_function(examples):