File size: 2,484 Bytes
a7e77b8 70a73a6 43da070 70a73a6 a7e77b8 43da070 a7e77b8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
import gradio as gr
from transformers import pipeline
import spaces
pipe = pipeline("text-generation", model="fava-uw/fava-model")
@spaces.GPU()
def chat(refrence: str, passage: str) -> str:
"""
Help to chat with fava
Args:
refrence: Pass the refrence text here
passage: main passage here
"""
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'].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;'>")
output = output.replace("</delete>", "</span>")
output = output.replace("<entity>", "<span style='background-color: #E9A2D9; border-bottom: 1px dotted;'></span>")
output = output.replace("<relation>", "<span style='background-color: #F3B78B; border-bottom: 1px dotted;'></span>")
output = output.replace("<contradictory>", "<span style='background-color: #FFFF9B; border-bottom: 1px dotted;'></span>")
output = output.replace("<unverifiable>", "<span style='background-color: #D3D3D3; border-bottom: 1px dotted;'></span><u>")
output = output.replace("<invented>", "<span style='background-color: #BFE9B9; border-bottom: 1px dotted;'></span>")
output = output.replace("<subjective>", "<span style='background-color: #D3D3D3; border-bottom: 1px dotted;'></span><u>")
output = output.replace("</entity>", "")
output = output.replace("</relation>", "")
output = output.replace("</contradictory>", "")
output = output.replace("</unverifiable>", "</u>")
output = output.replace("</invented>", "")
output = output.replace("</subjective>", "</u>")
output = output.replace("Edited:", "")
return f'<div style="font-weight: normal;">{output}</div>';
with gr.Blocks() as demo:
gr.Markdown("# ExMC")
with gr.Row():
refrence = gr.Textbox(label="Refrence", placeholder="Enter your Refrence here")
passage = gr.Textbox(label="Passage", placeholder="Enter your passage")
submit_btn = gr.Button("Submit")
submit_btn.click(fn=chat, inputs=[refrence, passage], outputs=gr.HTML("Output result"))
demo.launch() |