Update app.py

app.py

@@ -11,14 +11,16 @@ model = AutoModelForSequenceClassification.from_pretrained(model_name).to(device
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 
-# tempalte
+# template
 template = Template(
 """
 # Document:
-{{ text }}
+{{ document }}
 
 # Conversation:
-{{ conversation }}
+{% for message in conversation %}
+{{ message.role }}: {{ message.content }}
+{% endfor %}
 """
 )