iamomtiwari commited on
Commit
786d2c0
·
verified ·
1 Parent(s): 4cfa6fe

Create pipeline.py

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