Update app.py

app.py

@@ -13,6 +13,7 @@ INPUT = "Read the following references:\n{evidence}\nPlease identify all the err
 model = vllm.LLM(model="fava-uw/fava-model") 
 def result(passage, reference):
     prompt = [INPUT.format_map({"evidence":reference, "output":passage})]
+    print(prompt)
     sampling_params = vllm.SamplingParams(
         temperature=0,
         top_p=1.0,
@@ -33,9 +34,9 @@ def result(passage, reference):
     output = output.replace("</entity>", "")
     output = output.replace("</relation>", "")
     output = output.replace("</contradictory>", "")
-    output = output.replace("</u></unverifiable>", "")
+    output = output.replace("</unverifiable>", "</u>")
     output = output.replace("</invented>", "")
-    output = output.replace("</u></subjective>", "")
+    output = output.replace("</subjective>", "</u>")
     output = output.replace("Edited:", "")
     return f'<div style="font-weight: normal;">{output}</div>'; #output;