rajkumarrrk commited on
Commit
760c424
·
verified ·
1 Parent(s): 8f663c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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)