Joshua Lochner
commited on
Commit
·
dd33b0c
1
Parent(s):
1fc070a
Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -11,6 +11,7 @@ from transformers import (
|
|
| 11 |
)
|
| 12 |
from typing import Any, Dict, List
|
| 13 |
import os
|
|
|
|
| 14 |
|
| 15 |
CATEGORIES = [None, 'SPONSOR', 'SELFPROMO', 'INTERACTION']
|
| 16 |
|
|
@@ -287,7 +288,6 @@ class PreTrainedPipeline():
|
|
| 287 |
model=self.model2, tokenizer=self.tokenizer2)
|
| 288 |
|
| 289 |
def __call__(self, inputs: str)-> List[Dict[str, Any]]:
|
| 290 |
-
|
| 291 |
|
| 292 |
if ' ' not in inputs and inputs.count(',') >= 2: # Automated call (compressed string)
|
| 293 |
split_info = inputs.split(',', 1)
|
|
|
|
| 11 |
)
|
| 12 |
from typing import Any, Dict, List
|
| 13 |
import os
|
| 14 |
+
import numpy as np
|
| 15 |
|
| 16 |
CATEGORIES = [None, 'SPONSOR', 'SELFPROMO', 'INTERACTION']
|
| 17 |
|
|
|
|
| 288 |
model=self.model2, tokenizer=self.tokenizer2)
|
| 289 |
|
| 290 |
def __call__(self, inputs: str)-> List[Dict[str, Any]]:
|
|
|
|
| 291 |
|
| 292 |
if ' ' not in inputs and inputs.count(',') >= 2: # Automated call (compressed string)
|
| 293 |
split_info = inputs.split(',', 1)
|