diff --git a/allegro.py b/allegro.py index b6bbba7..65a5580 100644 --- a/allegro.py +++ b/allegro.py @@ -26,15 +26,19 @@ def fetch_documents(): response = collection.query.fetch_objects() documents = [] for item in response["data"]["Get"]["Document"]: - file_name = item.get("fileName", "unknown_file") - content = item.get("content", "") - if content: - documents.append(f"fileName: {file_name}, content: {content}") - print(f"fileName: {file_name}") - return documents + print(item) + #file_name = item.get("fileName", "unknown_file") + #content = item.get("content", "") + #if content: + # documents.append(f"fileName: {file_name}, content: {content}") + # print(f"fileName: {file_name}") + #return documents + #return documents -documents = fetch_documents() +fetch_documents() + +""" # 3️⃣ Inicjalizacja modelu model_name = "allegro/multislav-5lang" device = "cuda" if torch.cuda.is_available() else "cpu" @@ -99,3 +103,4 @@ model.save_pretrained("./models/allegro") tokenizer.save_pretrained("./models/allegro") print("✅ Model został wytrenowany i zapisany!") +""" \ No newline at end of file