improve logs

src/app.py

@@ -72,7 +72,6 @@ def on_show_prompt_click(criteria, context, user_message, assistant_message, sta
 
     messages = get_messages(test_case=test_case, sub_catalog_name=state['selected_sub_catalog'])
     prompt = get_prompt(messages, criteria_name)
-    prompt = json.dumps(prompt, indent=4)
     prompt = prompt.replace('<', '&lt;').replace('>', '&gt;').replace('\\n', '<br>')
     return gr.Markdown(prompt)
 

src/model.py

@@ -63,7 +63,7 @@ def get_prompt(messages, criteria_name):
 
 
 def generate_text(messages, criteria_name):
-    logger.debug(f'Prompts content is: \n{messages}')
+    logger.debug(f'Messages are: \n{messages}')
     
     mock_model_call = os.getenv('MOCK_MODEL_CALL') == 'true'
     if mock_model_call:
@@ -73,7 +73,8 @@ def generate_text(messages, criteria_name):
     
     start = time()
     chat = get_prompt(messages, criteria_name)
-
+    logger.debug(f'Prompt is \n{chat}')
+    
     with torch.no_grad():
         output = model.generate(chat, sampling_params, use_tqdm=False)