diff --git a/lib/model_prompts.json b/model_prompts.json similarity index 100% rename from lib/model_prompts.json rename to model_prompts.json diff --git a/ollama.py b/ollama.py index 6e8e96ff..cd535e88 100644 --- a/ollama.py +++ b/ollama.py @@ -86,7 +86,6 @@ def unload(model): def generate(model, prompt): # auto load error = load(model) - print(error) if error is not None: return error generated = llms[model]( diff --git a/lib/template.py b/template.py similarity index 92% rename from lib/template.py rename to template.py index 7839508b..e03c2feb 100644 --- a/lib/template.py +++ b/template.py @@ -1,7 +1,7 @@ from difflib import SequenceMatcher import json -with open("model_prompts.json", "r") as f: +with open("./model_prompts.json", "r") as f: model_prompts = json.load(f)