Browse Source

fix build

Bruce MacDonald 1 year ago
parent
commit
a12df7faf4
3 changed files with 1 additions and 2 deletions
  1. 0 0
      model_prompts.json
  2. 0 1
      ollama.py
  3. 1 1
      template.py

+ 0 - 0
lib/model_prompts.json → model_prompts.json


+ 0 - 1
ollama.py

@@ -86,7 +86,6 @@ def unload(model):
 def generate(model, prompt):
 def generate(model, prompt):
     # auto load
     # auto load
     error = load(model)
     error = load(model)
-    print(error)
     if error is not None:
     if error is not None:
         return error
         return error
     generated = llms[model](
     generated = llms[model](

+ 1 - 1
lib/template.py → template.py

@@ -1,7 +1,7 @@
 from difflib import SequenceMatcher
 from difflib import SequenceMatcher
 import json
 import json
 
 
-with open("model_prompts.json", "r") as f:
+with open("./model_prompts.json", "r") as f:
     model_prompts = json.load(f)
     model_prompts = json.load(f)