Sakalti commited on
Commit
4012ee4
1 Parent(s): 7b9f968

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ if st.button("トレーニング開始"):
33
  # データセットの準備
34
  st.write("データセットのロード中...")
35
  from datasets import load_dataset
36
- dataset = load_dataset(dataset_name, split="train")
37
 
38
  # トレーニング用のデータセットの準備
39
  def tokenize_function(examples):
 
33
  # データセットの準備
34
  st.write("データセットのロード中...")
35
  from datasets import load_dataset
36
+ dataset = load_dataset(dataset_name, split="swahili")
37
 
38
  # トレーニング用のデータセットの準備
39
  def tokenize_function(examples):