Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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,
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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])
|