shhossain commited on
Commit
6d9533b
1 Parent(s): 9706db3

Delete smodelpipeline.py

Browse files
Files changed (1) hide show
  1. 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