update app.py

app.py

@@ -33,19 +33,22 @@ def translate(inputs, task):
 
     #text = pipe(inputs, batch_size=BATCH_SIZE, generate_kwargs={"task": task}, return_timestamps=True)["text"]
     ts_model = WhisperModel(MODEL_NAME, device = device, compute_type = "int8")
-    text = ts_model.transcribe(inputs, task = "translate")
+    segments, _ = ts_model.transcribe(inputs, task = "translate")
+    lst = ''
+    for segment in segments:
+         lst = lst + segment.text
 
-    encoded_text = tokenizer(text, return_tensors="pt")
+    encoded_text = tokenizer(lst, return_tensors="pt")
     tokenizer.src_lang = "en_XX"
     
     generated_tokens = model.generate(
         **encoded_text,
         forced_bos_token_id=tokenizer.lang_code_to_id["fr_XX"]
     )
+    
     result = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
     return result
-        
-
+       
 demo = gr.Blocks()
 
 mf_transcribe = gr.Interface(