diff --git a/requirements.txt b/requirements.txt
index eaa6c81..dc90de6 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,5 +8,5 @@ pydantic
 sqlmodel
 pytest
 httpx
-numpy<2
+numpy>=2.0.0
 ollama
\ No newline at end of file
diff --git a/src/main.py b/src/main.py
index 5bb632b..17d99e8 100644
--- a/src/main.py
+++ b/src/main.py
@@ -3,6 +3,7 @@
 from commonforms import prepare_form 
 from pypdf import PdfReader
 from controller import Controller
+from typing import Union
 
 def input_fields(num_fields: int):
     fields = []