From 32b17c337350b47e4a714d1db3f6e5417f38f8ca Mon Sep 17 00:00:00 2001 From: martin legrand Date: Sat, 29 Mar 2025 13:17:01 +0100 Subject: [PATCH] feat : better logging --- server/app.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/server/app.py b/server/app.py index 2e7a1d9..d9e7860 100644 --- a/server/app.py +++ b/server/app.py @@ -15,7 +15,7 @@ app = Flask(__name__) assert args.provider in ["ollama", "llamacpp"], f"Provider {args.provider} does not exists. see --help for more information" -generator = OllamaLLM if args.provider == "ollama" else LlamacppLLM +generator = OllamaLLM() if args.provider == "ollama" else LlamacppLLM() @app.route('/generate', methods=['POST']) def start_generation():