Update app.py

app.py

@@ -16,7 +16,7 @@ def chat(refrence: str, passage: str) -> str:
     text = f"Read the following references:\n{refrence}\nPlease identify all the errors in the following text using the information in the references provided and suggest edits if necessary:\n[Text] {passage}\n[Edited] "
     outputs = pipe(text)
     print(outputs)
-    output = outputs[0]['generated_text']
+    output = outputs[0]['generated_text'].split("[Edited]")[1]
     output = output.replace("<mark>", "<span style='color: green; font-weight: bold;'> ")
     output = output.replace("</mark>", " </span>")
     output = output.replace("<delete>", "<span style='color: red; text-decoration: line-through;'>")
@@ -42,7 +42,6 @@ with gr.Blocks() as demo:
         refrence = gr.Textbox(label="Refrence", placeholder="Enter your Refrence here")
         passage = gr.Textbox(label="Passage", placeholder="Enter your passage")
         submit_btn = gr.Button("Submit")
-    output_text = gr.Textbox(label="Result")
-    submit_btn.click(fn=chat, inputs=[refrence, passage], outputs=output_text)
+    submit_btn.click(fn=chat, inputs=[refrence, passage], outputs=gr.HTML("Output result"))
 
 demo.launch()