llava als model zum testen

This commit is contained in:
2024-10-17 21:49:26 +02:00
parent 761a20606a
commit ece63f511b

View File

@@ -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: