Debug

app.py

@@ -150,6 +150,8 @@ def chat_with_model(messages):
     inputs = current_tokenizer(prompt, return_tensors="pt").to(device)
     prompt_len = inputs["input_ids"].shape[-1]
 
+    print(prompt)
+
     streamer = RichTextStreamer(
         tokenizer=current_tokenizer,
         prompt_len=prompt_len,
@@ -172,6 +174,8 @@ def chat_with_model(messages):
     updated_messages = messages.copy()
     updated_messages.append({"role": "assistant", "content": ""})
 
+    print(updated_messages)
+
     for token_info in streamer:
         token_str = token_info["token"]
         token_id = token_info["token_id"]