Update app.py

app.py

@@ -1,40 +1,29 @@
-from transformers import pipeline, AutoTokenizer, AutoModelWithLMHead, TranslationPipeline
+from transformers import pipeline
 import gradio as gr
+import os
+import deepl
+import openai
 
+TARGET_LANG = "EN-GB"
+deepl_key = os.environ.get('DEEPL')
 
-
+translator = deepl.Translator(deepl_key)
 pipe = pipeline(model="torileatherman/train_first_try")  # change to "your-username/the-name-you-picked"
 
 def transcribe(audio):
-    text = pipe(audio)["text"]
-    return text
-
-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), 
-                                            device=0)
-
-def translate(text):
-    translation = translation_pipeline([text], max_length=512)
-    return translation
-
-demo = gr.Blocks()
-
-with demo:
-    
-    title="Whisper Small Swedish",
-    description="Realtime demo for Swedish speech recognition using a fine-tuned Whisper small model."
-    
-    inputs_audio = gr.Audio(source="microphone", type="filepath"), 
- 
-    text = gr.Textbox()
-    translation = gr.Label()
-
-    b1 = gr.Button("Record audio")
-    b2 = gr.Button("Translate text")
-
-    b1.click(transcribe, inputs=inputs_audio, outputs=text)
-    b2.click(translate, inputs=text, outputs=translation)
-
-demo.launch()
+    text_sv = pipe(audio)["text"]
+    print(f"Audio transcribed: {text_sv}")
+    text_en = translator.translate_text(text_sv, target_lang=TARGET_LANG).text
+    print(f"Text translated: {text_en}")
+    return text_sv, text_en
+
+iface = gr.Interface(
+    fn=transcribe,
+    inputs=gr.Audio(source="microphone", type="filepath"),
+    outputs=[gr.Textbox(label="Transcribed text"),
+             gr.Textbox(label="English translation")],
+    title="Swedish speech to english text",
+    description="Transcribing swedish speech to text and translating to english!",
+)
+
+iface.launch()