diff --git a/allegro.py b/allegro.py
index 94e3a35..2ee14c1 100644
--- a/allegro.py
+++ b/allegro.py
@@ -36,12 +36,8 @@ def fetch_documents():
     return documents
     #return documents
 
+documents = fetch_documents()
 
-fetch_documents()
-
-client.close()
-
-"""
 # 3️⃣ Inicjalizacja modelu
 model_name = "allegro/multislav-5lang"
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -105,5 +101,4 @@ trainer.train()
 model.save_pretrained("./models/allegro")
 tokenizer.save_pretrained("./models/allegro")
 
-print("✅ Model został wytrenowany i zapisany!")
-"""
\ No newline at end of file
+print("✅ Model został wytrenowany i zapisany!")
\ No newline at end of file