johann22 commited on
Commit
90a1263
·
1 Parent(s): d044f1b

Update app_dialogue.py

Browse files
Files changed (1) hide show
  1. app_dialogue.py +1 -1
app_dialogue.py CHANGED
@@ -552,7 +552,7 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
552
 
553
  query = prompt_list_to_tgi_input(formated_prompt_list)
554
  #stream = client.generate_stream(prompt=query, **generation_args)
555
- stream = client.text_generation(query, **generation_args, stream=True)
556
  acc_text = ""
557
  for idx, response in enumerate(stream):
558
  text_token = response.token.text
 
552
 
553
  query = prompt_list_to_tgi_input(formated_prompt_list)
554
  #stream = client.generate_stream(prompt=query, **generation_args)
555
+ stream = client.text_generation(query, **generation_args, stream=True,details=True, return_full_text=True)
556
  acc_text = ""
557
  for idx, response in enumerate(stream):
558
  text_token = response.token.text