Update app_dialogue.py

app_dialogue.py

@@ -551,8 +551,8 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
             )
 
         query = prompt_list_to_tgi_input(formated_prompt_list)
-        stream = client.generate_stream(prompt=query, **generation_args)
-
+        #stream = client.generate_stream(prompt=query, **generation_args)
+        stream = client.text_generation(query, **generate_args, stream=True)
         acc_text = ""
         for idx, response in enumerate(stream):
             text_token = response.token.text
@@ -637,7 +637,9 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
                         break
 
         query = prompt_list_to_tgi_input(formated_prompt_list)
-        generated_text = client.generate(prompt=query, **generation_args).generated_text
+        generated_text = client.text_generation(query, **generation_args, stream=False)
+
+        #generated_text = client.generate(prompt=query, **generation_args).generated_text
         if generated_text.endswith("\nUser"):
             generated_text = generated_text[:-5]