Add loguru

model.py

@@ -4,6 +4,7 @@ from typing import Iterator
 
 from llama_cpp import Llama
 from huggingface_hub import hf_hub_download
+from loguru import logger
 
 def download_model():
     print(f"Downloading model")
@@ -32,6 +33,8 @@ def load_llm():
 
 llm = load_llm()
 
+logger.info(f"done load llm")
+
 def get_prompt(message: str, chat_history: list[tuple[str, str]],
                system_prompt: str):
     prompt=""
@@ -60,6 +63,9 @@ def run(message: str,
 
     prompt = get_prompt(message, chat_history, system_prompt)
 
+    logger.info(f"New prompt")
+    logger.info(f"{prompt}")
+
     #stop=["</s>"]
     stop = ["USER:", "ASSISTANT:"]
     outputs = []