Create app.py

app.py

@@ -0,0 +1,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)
+    output_texts = [it.outputs[0].text for it in outputs]
+    output = outputs[0].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")
+    output_text = gr.Textbox(label="Result")
+    submit_btn.click(fn=chat, inputs=[refrence, passage], outputs=output_text)
+
+demo.launch()