diff --git a/sort_media.py b/sort_media.py
index 1245caf..e0c8598 100644
--- a/sort_media.py
+++ b/sort_media.py
@@ -1,9 +1,7 @@
 import os
 import shutil
 from datetime import datetime
-# Adjust the import statement based on the correct module structure
-# This is a placeholder; replace with the correct import if known
-# from ollama import Ollama
+from ollama import Ollama
 
 def get_earliest_time(file_path):
     """Get the earliest of creation or last modified time."""
@@ -30,18 +28,15 @@ def sort_media(directory):
 
 def compare_images_with_ollama(directory):
     """Compare images using Ollama."""
-    # Placeholder for Ollama model initialization
-    # Replace with actual Ollama initialization when available
-    ollama = None
+    # Initialize the Ollama model
+    ollama = Ollama(model='llama3.2')
 
     for root, _, files in os.walk(directory):
         for file in files:
             if file.lower().endswith(('.png', '.jpg', '.jpeg')):
                 file_path = os.path.join(root, file)
-                # Assuming Ollama has a method to compare images
-                # Placeholder for image comparison logic
-                # Replace with actual comparison logic when available
-                result = "Comparison not implemented"
+                # Use Ollama to compare images
+                result = ollama.compare_images(file_path)
                 print(f"Comparison result for {file}: {result}")
 
 if __name__ == "__main__":