all add
Browse files
app.py
CHANGED
|
@@ -30,7 +30,6 @@ model.config.label2id = {label: i for i, label in zip(range(num_labels), unsmile
|
|
| 30 |
pipe = TextClassificationPipeline(
|
| 31 |
model = model,
|
| 32 |
tokenizer = tokenizer,
|
| 33 |
-
device=0,
|
| 34 |
return_all_scores=True,
|
| 35 |
function_to_apply='sigmoid'
|
| 36 |
)
|
|
|
|
| 30 |
pipe = TextClassificationPipeline(
|
| 31 |
model = model,
|
| 32 |
tokenizer = tokenizer,
|
|
|
|
| 33 |
return_all_scores=True,
|
| 34 |
function_to_apply='sigmoid'
|
| 35 |
)
|