diff --git a/aider_gitea/__main__.py b/aider_gitea/__main__.py
index a99f791..1e2c10a 100644
--- a/aider_gitea/__main__.py
+++ b/aider_gitea/__main__.py
@@ -62,6 +62,8 @@ def parse_args():
 def main():
     logging.basicConfig(level='INFO')
     args = parse_args()
+    aider_gitea.CODE_MODEL = args.code_model
+    aider_gitea.EVALUATOR_MODEL = args.evaluator_model
 
     seen_issues_db = SeenIssuesDB()
     client = GiteaClient(args.gitea_url, secrets.gitea_token())