Update app.py

app.py

@@ -9,8 +9,6 @@ huggingface_hub.login(token=os.environ["HF_TOKEN"])
 # Fava prompt
 INPUT = "Read the following references:\n{evidence}\nPlease identify all the errors in the following text using the information in the references provided and suggest edits if necessary:\n[Text] {output}\n[Edited] "
 
-_article = f'<center><img src="https://huggingface.co/spaces/fava-uw/fava/resolve/main/images/fava.png" width=800px></center>'
-
 model = vllm.LLM(model="fava-uw/fava-model") 
 def result(passage, reference):
     prompt = [INPUT.format_map({"evidence":reference, "output":passage})]
@@ -41,7 +39,10 @@ def result(passage, reference):
     return f'<div style="font-weight: normal;">{output}</div>'; #output;
 
 if __name__ == "__main__":
-    # description="Enter some text and a reference for the text and FAVA will detect and edit any hallucinations present."
-    custom_footer = """<div style="text-align: center; padding: 20px;"><img src="./images/fava.png" alt="Example Image"></div>"""
-    demo = gradio.Interface(fn=result, inputs=["text", "text"], outputs="html", title="Fine-grained Hallucination Detection and Editing (FAVA)", description=custom_footer)
+    article = """<center><img src="/fava-uw/fava/images/fava.png" alt="Example Image"></center>"""
+    examples = [["Canada's oldest national park, Banff, was established in 1886. It recently won a Nature's Choice 2023 award for its beautiful mountainous terrain. It's the best national park ever.", 
+                 "Banff National Park is Canada's oldest national park, established in 1885 as Rocky Mountains Park. Located in Alberta's Rocky Mountains, 110–180 kilometres (68–112 mi) west of Calgary, Banff encompasses 6,641 square kilometres (2,564 sq mi) of mountainous terrain."]]
+    demo = gradio.Interface(fn=result, 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.", article=article,
+                           examples=examples, allow_flagging="manual", flagging_options=["wrong detection", "wrong editing", "both wrong", "other"], flagging_dir="flags")
     demo.launch(share=True)