Upload handler.py

handler.py

@@ -82,9 +82,9 @@ def generate(
     end_sequences: list[str] = ["[INST]", "[/INST]", "\n"]
 ) -> dict:
     
-    print(instruction)
-    print(conclusions)
-    print(context)
+    LOGGER.info(f"instruction: {instruction}")
+    LOGGER.info(f"conclusions: {conclusions}")
+    LOGGER.info(f"context: {context}")
     # Check if the system_prompt is provided, else construct it from instruction, conclusions, and context
     if not system_prompt and instruction is not None and conclusions is not None and context is not None:
         system_prompt = "Instruction: {}\nConclusions:\n".format(instruction)