Luigi commited on
Commit
5fc0117
·
1 Parent(s): 960db60

add prompt preview for debug

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -167,6 +167,7 @@ def chat_response(user_msg, chat_history, system_prompt,
167
 
168
  pipe = load_pipeline(model_name)
169
  prompt = format_conversation(history, enriched, pipe.tokenizer)
 
170
  streamer = TextIteratorStreamer(pipe.tokenizer,
171
  skip_prompt=True,
172
  skip_special_tokens=True)
 
167
 
168
  pipe = load_pipeline(model_name)
169
  prompt = format_conversation(history, enriched, pipe.tokenizer)
170
+ debug += f"\n\n--- Prompt Preview ---\n```\n{prompt}\n```"
171
  streamer = TextIteratorStreamer(pipe.tokenizer,
172
  skip_prompt=True,
173
  skip_special_tokens=True)