Update app.py

app.py

@@ -28,18 +28,18 @@ def result(passage, reference):
     output = output.replace("<entity>", "<span style='background-color: #E9A2D9; border-bottom: 1px dotted;'>entity</span>")
     output = output.replace("<relation>", "<span style='background-color: #F3B78B; border-bottom: 1px dotted;'>relation</span>")
     output = output.replace("<contradictory>", "<span style='background-color: #FFFF9B; border-bottom: 1px dotted;'>contradictory</span>")
-    output = output.replace("<unverifiable>", "<span style='background-color: #B2DCE5; border-bottom: 1px dotted;'>unverifiable</span>")
+    output = output.replace("<unverifiable>", "<span style='background-color: #D3D3D3; border-bottom: 1px dotted;'>unverifiable</span><u>")
     output = output.replace("<invented>", "<span style='background-color: #BFE9B9; border-bottom: 1px dotted;'>invented</span>")
-    output = output.replace("<subjective>", "<span style='background-color: #B5B7F5; border-bottom: 1px dotted;'>subjective</span>")
+    output = output.replace("<subjective>", "<span style='background-color: #D3D3D3; border-bottom: 1px dotted;'>subjective</span><u>")
     output = output.replace("</entity>", "")
     output = output.replace("</relation>", "")
     output = output.replace("</contradictory>", "")
-    output = output.replace("</unverifiable>", "")
+    output = output.replace("</u></unverifiable>", "")
     output = output.replace("</invented>", "")
-    output = output.replace("</subjective>", "")
+    output = output.replace("</u></subjective>", "")
     output = output.replace("Edited:", "")
     return f'<div style="font-weight: normal;">{output}</div>'; #output;
 
 if __name__ == "__main__":
-    demo = gradio.Interface(fn=result, inputs=["text", "text"], outputs="html")
+    demo = gradio.Interface(fn=result, flagging_dir="flagged/", inputs=["text", "text"], outputs="html", title="Fine-grained Hallucination Detection and Editing (FAVA)", description="Enter some text and a reference for the text and FAVA will detect and edit any hallucinations present.")
     demo.launch(share=True)