johann22 commited on
Commit
9ea8d9c
·
1 Parent(s): 8f8e956

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
- resp = client.text_generation(query, **generation_args, stream=True,details=True, return_full_text=True)
556
  stream = ""
557
  for response in resp:
558
  stream += response.token.text
 
552
 
553
  query = prompt_list_to_tgi_input(formated_prompt_list)
554
  #stream = client.generate_stream(prompt=query, **generation_args)
555
+ resp = client.text_generation(query, **generation_args, stream=True,details=True, return_full_text=False)
556
  stream = ""
557
  for response in resp:
558
  stream += response.token.text