Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 from jinja2 import Template
 import torch
+import json
 
 
 # load the judge
@@ -27,6 +28,7 @@ template = Template(
 
 def judge_reliability(document: str, conversation: str):
     with torch.no_grad():
+        conversation = json.loads(conversation)
         text = template.render(text=document, conversation=conversation)
         encoded = tokenizer([text], padding=True)
         input_ids = torch.tensor(encoded.input_ids).to(device)