diff --git a/sort_media.py b/sort_media.py
index cc8d6e4..f1c7585 100644
--- a/sort_media.py
+++ b/sort_media.py
@@ -29,7 +29,7 @@ def sort_media(directory):
 def compare_images_with_ollama(directory):
     """Compare images using Ollama."""
     # Initialize the Ollama model
-    model = 'llama3.2'
+    model = 'llava-llama3:8b'
 
     for root, _, files in os.walk(directory):
         for file in files: