IBounhas commited on
Commit
d445114
·
1 Parent(s): f45632f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -67,6 +67,7 @@ model.to(device)
67
  def predict_similarity(sentence_pair):
68
  test_input = tokenizer(sentence_pair, padding='max_length', max_length = param_max_length, truncation=True, return_tensors="pt").to(device)
69
  test_input['input_ids'] = test_input['input_ids']
 
70
  test_input['attention_mask'] = test_input['attention_mask']
71
  del test_input['token_type_ids']
72
  output = model(test_input)
 
67
  def predict_similarity(sentence_pair):
68
  test_input = tokenizer(sentence_pair, padding='max_length', max_length = param_max_length, truncation=True, return_tensors="pt").to(device)
69
  test_input['input_ids'] = test_input['input_ids']
70
+ print(test_input['input_ids'])
71
  test_input['attention_mask'] = test_input['attention_mask']
72
  del test_input['token_type_ids']
73
  output = model(test_input)