Update app.py

app.py

@@ -17,8 +17,7 @@ def train_model(number):
     number = int(number)
     try:
         dataset = load_dataset("allenai/c4", "sk", split="train", streaming=True)
-        slovak_texts = list(dataset.take(number))
-        
+        slovak_texts = [example['text'] for example in dataset.take(number)]        
         inputs = tokenizer(
             slovak_texts,
             return_tensors="pt",