Update app.py

app.py

@@ -12,10 +12,10 @@ model_pipeline = pipeline(task="text-classification", model=model_name, device=d
 # templates
 conv_template = Template(
     """
-    # Document:
+    # Context:
     {{ document }}
-
-    # Conversation:
+    
+    # Claim:
     {% for message in conversation %}
     {{ message.role }}: {{ message.content }}
     {% endfor %}
@@ -24,24 +24,22 @@ conv_template = Template(
 
 qa_template = Template(
     """
-    # Document:
+    # Context:
     {{ document }}
-
-    # Question:
-    {{ question }}
-
-    # Answer:
-    {{ answer }}
+    
+    # Claim:
+    user: {{ question }}
+    assistant: {{ answer }}
     """
 )
 
 nli_template = Template(
     """
-    # Document:
+    # Context:
     {{ document }}
-
+    
     # Claim:
-    {{ claim }}
+    assistant: {{ claim }}
     """
 )