Update README.md

README.md

@@ -34,19 +34,19 @@ weighted avg       0.80      0.79      0.79       200
 For best results, we recommend starting with the following prompting strategy (and encourage tweaks as you see fit):
 
 ```python
-def format_input(query, response):
-    """Your query field can be a dialogue or a single query with optional context included"""
-    input = f"""Your job is to evaluate whether a machine learning model has hallucinated or not.
-    A hallucination occurs when the response is coherent but factually incorrect or nonsensical 
+def format_input(reference, query, response):
+    prompt = f"""Your job is to evaluate whether a machine learning model has hallucinated or not.
+    A hallucination occurs when the response is coherent but factually incorrect or nonsensical
     outputs that are not grounded in the provided context.
     You are given the following information:
-      ####INFO####
-      [Query]: {query}
-      [Model Response]: {response}
-      ####END INFO####
-      Based on the information provided is the model output a hallucination? Respond with only "yes" or "no"
-      """
-      return input
+    ####INFO####
+    [Knowledge]: {reference}
+    [User Input]: {query}
+    [Model Response]: {response}
+    ####END INFO####
+    Based on the information provided is the model output a hallucination? Respond with only "yes" or "no"
+    """
+    return input
 
 text = format_input(query='Based on the follwoing
           <context>Walrus are the largest mammal</context>