Delete smodelpipeline.py
Browse files- smodelpipeline.py +0 -23
smodelpipeline.py
DELETED
@@ -1,23 +0,0 @@
|
|
1 |
-
from transformers import Pipeline
|
2 |
-
from sentence_transformers import SentenceTransformer
|
3 |
-
import torch
|
4 |
-
|
5 |
-
class SentimentModelPipe(Pipeline):
|
6 |
-
|
7 |
-
def __init__(self, **kwargs):
|
8 |
-
Pipeline.__init__(self, **kwargs)
|
9 |
-
self.smodel = SentenceTransformer(kwargs.get("embedding_model", "sentence-transformers/all-MiniLM-L6-v2"))
|
10 |
-
|
11 |
-
def _sanitize_parameters(self, **kw):
|
12 |
-
return {}, {}, {}
|
13 |
-
|
14 |
-
def preprocess(self, inputs):
|
15 |
-
return self.smodel.encode(inputs, convert_to_tensor=True)
|
16 |
-
|
17 |
-
def postprocess(self, outputs):
|
18 |
-
return outputs.argmax(1).item()
|
19 |
-
|
20 |
-
def _forward(self, tensor):
|
21 |
-
with torch.no_grad():
|
22 |
-
out = self.model(tensor)
|
23 |
-
return out
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|