Update app.py

app.py

@@ -1,4 +1,4 @@
-from transformers import pipeline
+from transformers import pipeline, AutoTokenizer, AutoModelWithLMHead, TranslationPipeline
 import gradio as gr
 import os
 
@@ -8,7 +8,7 @@ def transcribe(audio):
     text = pipe(audio)["text"]
     return text
 
-translation_pipeline = TranslationPipeline( model=AutoModelWithLMHead.from_pretrained("SEBIS/legal_t5_small_trans_sv_en"),
+translation_pipeline = TranslationPipeline(model=AutoModelWithLMHead.from_pretrained("SEBIS/legal_t5_small_trans_sv_en"),
                                             tokenizer=AutoTokenizer.from_pretrained(pretrained_model_name_or_path = "SEBIS/legal_t5_small_trans_sv_en", 
                                             do_lower_case=False, 
                                             skip_special_tokens=True),