|
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() |
