Update app.py

direct return of final result

app.py

@@ -432,7 +432,7 @@ with gr.Blocks() as demo:
                             tgt_index = int(tgt_index)
                         )
                         run_edit_text()
-                        return 0
+                        return load_pil_img()
                         
                     add_button.click(run_edit_text_wrapper, 
                         inputs = [num_tokens_global,
@@ -443,7 +443,7 @@ with gr.Blocks() as demo:
                                     tgt_prompt ,
                                     tgt_index
                                 ],        
-                        outputs = [],queue=True,
+                        outputs = [canvas_text_edit],queue=True,
                     )
                     
                     load_button  = gr.Button("Load results")