From 04747ff17b19f21afedc05bf11755f88ede3173e Mon Sep 17 00:00:00 2001
From: "l.gabrysiak" <l.gabrysiak@osadkowski.pl>
Date: Fri, 28 Feb 2025 21:19:50 +0100
Subject: [PATCH] mod allegro

---
 allegro.py | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

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