Update prompt

This commit is contained in:
Keannu Bernasol 2024-12-19 00:12:11 +08:00
parent c035671f3e
commit 8655a79e8b

View file

@ -105,7 +105,7 @@ class PDFHandler(FileSystemEventHandler):
prompt = f""" prompt = f"""
Read the text from the image and provide a category. Return as JSON. Read the text from the image and provide a category. Return as JSON.
Possible categories are: {possible_categories} Possible categories are: {possible_categories}. You are free to create a new one if none are suitable.
""" """
response = client.chat( response = client.chat(
model=get_secret("OLLAMA_MODEL"), model=get_secret("OLLAMA_MODEL"),