mod allegro

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!")
-"""
+print("✅ Model został wytrenowany i zapisany!")