iamomtiwari commited on
Commit
cd01a54
·
verified ·
1 Parent(s): 35a8dfb

Create pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +17 -0
pipeline.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import T5ForConditionalGeneration, T5Tokenizer
2
+ from googletrans import Translator
3
+ from transformers import pipeline
4
+
5
+ class GOTSummarizer:
6
+ def __init__(self):
7
+ self.model = T5ForConditionalGeneration.from_pretrained("iamomtiwari/NLP-Model")
8
+ self.tokenizer = T5Tokenizer.from_pretrained("iamomtiwari/NLP-Model")
9
+ self.translator = Translator()
10
+
11
+ def summarize(self, text, max_length=150):
12
+ inputs = self.tokenizer("summarize: " + text, return_tensors="pt", truncation=True)
13
+ outputs = self.model.generate(**inputs, max_length=max_length)
14
+ return self.tokenizer.decode(outputs[0], skip_special_tokens=True)
15
+
16
+ def translate(self, text, lang='hi'):
17
+ return self.translator.translate(text, dest=lang).text