diff --git a/code/main.py b/code/main.py
index a426d03..64351e7 100755
--- a/code/main.py
+++ b/code/main.py
@@ -50,13 +50,14 @@ IMPORTANT:
 
 
 def analyze_pdf(pdf_path):
+    model = os.environ.get("OPENROUTER_MODEL", "openai/gpt-5.4-mini")
+
     with open(pdf_path, 'rb') as f:
         pdf_data = f.read()
-
     base64_string = base64.b64encode(pdf_data).decode("utf-8")
 
     response = client.chat.completions.create(
-        model="openai/gpt-4o",
+        model=model,
         messages=[
             {
                 "role": "system",