From a52e27920a0789163b20d782c3c09b4f6d73839c Mon Sep 17 00:00:00 2001
From: "l.gabrysiak" <l.gabrysiak@osadkowski.pl>
Date: Tue, 25 Feb 2025 22:59:19 +0100
Subject: [PATCH] mod

---
 hft.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hft.py b/hft.py
index 7f4292f..62c63e6 100644
--- a/hft.py
+++ b/hft.py
@@ -207,7 +207,7 @@ def main():
         print("\nBrak danych do treningu!")
         return
 
-    dataset = Dataset.from_list(data)
+    dataset = Dataset.from_dict({k: [dic[k] for dic in data] for k in data[0]})
 
     def tokenize_function(examples):
         tokenized = tokenizer(