liujch1998 commited on
Commit
24bf24b
Β·
1 Parent(s): 62bce9d
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -87,7 +87,7 @@ class Interactive:
87
  'logit_calibrated': 0.0,
88
  'score': 0.5,
89
  'score_calibrated': 0.5,
90
- } for _ in statements]
91
  tok = self.tokenizer.batch_encode_plus(statements, return_tensors='pt', padding='longest')
92
  input_ids = tok.input_ids.to(device)
93
  attention_mask = tok.attention_mask.to(device)
 
87
  'logit_calibrated': 0.0,
88
  'score': 0.5,
89
  'score_calibrated': 0.5,
90
+ } for statement in statements]
91
  tok = self.tokenizer.batch_encode_plus(statements, return_tensors='pt', padding='longest')
92
  input_ids = tok.input_ids.to(device)
93
  attention_mask = tok.attention_mask.to(device)