Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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)
|