Update README.md

should return the formatted prompt instead of the input ? :)

README.md

@@ -46,7 +46,7 @@ def format_input(reference, query, response):
     ####END INFO####
     Based on the information provided is the model output a hallucination? Respond with only "yes" or "no"
     """
-    return input
+    return prompt
 
 text = format_input(reference="The apple mac has the best hardware",
           query="What computer has the best software?",