ilan541 commited on
Commit
abf5a81
·
1 Parent(s): 3f2d91b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -25,7 +25,11 @@ def get_probs(list_of_portions):
25
  y_pred_logits_0 = []
26
  y_pred_logits_1 = []
27
  for text in list_of_portions:
28
- inp = tokenizer(text, return_tensors='tf')
 
 
 
 
29
  y_pred = model(inp)
30
  y_pred_logits_0.append(y_pred.logits[:,0])
31
  y_pred_logits_1.append(y_pred.logits[:,1])
 
25
  y_pred_logits_0 = []
26
  y_pred_logits_1 = []
27
  for text in list_of_portions:
28
+ inp = tokenizer(text,
29
+ truncation=True,
30
+ padding='max_length',
31
+ max_length=512,
32
+ return_tensors='tf')
33
  y_pred = model(inp)
34
  y_pred_logits_0.append(y_pred.logits[:,0])
35
  y_pred_logits_1.append(y_pred.logits[:,1])