diff --git a/sort_media.py b/sort_media.py
index f1c7585..38228b5 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 = 'llava-llama3:8b'
+    model = 'llava'
 
     for root, _, files in os.walk(directory):
         for file in files: