mirror of
https://github.com/lemeow125/DocManagerBackend.git
synced 2025-01-18 17:13:00 +08:00
Update prompt
This commit is contained in:
parent
c035671f3e
commit
8655a79e8b
1 changed files with 1 additions and 1 deletions
|
@ -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"),
|
||||||
|
|
Loading…
Reference in a new issue