diff --git a/allegro.py b/allegro.py
index e0678ab..9e996c4 100644
--- a/allegro.py
+++ b/allegro.py
@@ -4,12 +4,11 @@ import weaviate
 import numpy as np
 from datasets import Dataset
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, TrainingArguments, Trainer, DataCollatorForSeq2Seq
-from weaviate.classes.config import Property, DataType, Configure
-from weaviate.classes.client import WeaviateClient
 from weaviate.classes.config import ConnectionParams
+import weaviate
 
 # 1️⃣ Połączenie z Weaviate
-client = WeaviateClient(
+client = weaviate.WeaviateClient(
     connection_params=ConnectionParams.from_params(
         http_host="weaviate",
         http_port=8080,