diff --git a/allegro.py b/allegro.py
index 54bc777..b7d0b99 100644
--- a/allegro.py
+++ b/allegro.py
@@ -1,11 +1,12 @@
 import os
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
-import torch
-import faiss
 from weaviate.connect import ConnectionParams
 import weaviate
 
+import torch
+import faiss
+import numpy as np
 from sentence_transformers import SentenceTransformer
 from datasets import Dataset
 from peft import LoraConfig, get_peft_model