Update handler.py

handler.py

@@ -168,7 +168,8 @@ class EndpointHandler():
         output_translations = []
         if self.model is not None:
             transcriptions = self.read_srt(transcript_path)
-            indic_translations = self.batch_translate(transcriptions, src_lang, tgt_lang)
+            trans_sents = [entry['text'] for entry in transcriptions]
+            indic_translations = self.batch_translate(trans_sents, src_lang, tgt_lang)
 
             for i in tqdm(range(len(transcriptions))):
                 entry = transcriptions[i]
@@ -177,18 +178,4 @@ class EndpointHandler():
 
             return output_translations
         else:
-            return []
-
-
-if __name__ == "__main__":
-    endpoint = EndpointHandler(quantization = "8-bit")
-    inputs = {}
-    inputs['src_lang'] = 'eng_Latn'
-    inputs['tgt_lang'] = 'tel_Telu'
-    inputs['transcript_path'] = './sample.srt'
-
-    outputs = endpoint.test(inputs)
-
-    print("Outputs: ")
-    for entry in outputs:
-        print(entry)
+            return []