Update app.py

app.py

@@ -24,7 +24,7 @@ def humanize(
     		length_penalty=length_penalty,
     		api_name="/paraphrase_text"
     )
-    return result
+    return ai_text, result
 
 with gr.Blocks() as demo:
     gr.Markdown("# Polygraf Writer")
@@ -45,8 +45,10 @@ with gr.Blocks() as demo:
                 label="Select Model Version",
             )
             process_button = gr.Button("Humanize Text")
+            gr.Markdown("### AI Generated article:")
+            ai_label = gr.HTML(label="AI")
             gr.Markdown("### Humanized article:")
-            output_label = gr.HTML(label="Output")
+            output_label = gr.HTML(label="Humanized")
 
         with gr.Column(scale=0.3):
                 temperature_slider = gr.Slider(minimum=0.5, maximum=2.0, step=0.1, value=1.2, label="Temperature")
@@ -86,7 +88,7 @@ with gr.Blocks() as demo:
                 top_k_slider,
                 length_penalty_slider,
             ],
-            outputs=output_label,
+            outputs=[ai_label, output_label],
         )
 
 if __name__ == "__main__":