Spaces:
Running
Running
rajkumarrrk
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -30,6 +30,7 @@ def judge_reliability(document: str, conversation: str):
|
|
30 |
with torch.no_grad():
|
31 |
conversation = json.loads(conversation)
|
32 |
text = template.render(text=document, conversation=conversation)
|
|
|
33 |
encoded = tokenizer([text], padding=True)
|
34 |
input_ids = torch.tensor(encoded.input_ids).to(device)
|
35 |
attention_mask = torch.tensor(encoded.attention_mask).to(device)
|
|
|
30 |
with torch.no_grad():
|
31 |
conversation = json.loads(conversation)
|
32 |
text = template.render(text=document, conversation=conversation)
|
33 |
+
print(text)
|
34 |
encoded = tokenizer([text], padding=True)
|
35 |
input_ids = torch.tensor(encoded.input_ids).to(device)
|
36 |
attention_mask = torch.tensor(encoded.attention_mask).to(device)
|