Update app.py

app.py

@@ -37,10 +37,10 @@ def split_text_into_sentences(text):
 @spaces.GPU(duration=120)
 # Translation function
 def translate(source_text, source_lang, target_lang, batch_size=16):
-    if source_lang== 'en' and target_lang == 'tr':
-       model_name = f"Helsinki-NLP/opus-mt-tc-big-en-tr"
+    if source_lang == 'en' and target_lang == 'tr':
+        model_name = f"Helsinki-NLP/opus-mt-tc-big-en-tr"
     else:
-       model_name = f"Helsinki-NLP/opus-mt-{source_lang}-{target_lang}"
+        model_name = f"Helsinki-NLP/opus-mt-{source_lang}-{target_lang}"
     tokenizer = MarianTokenizer.from_pretrained(model_name)
     model = MarianMTModel.from_pretrained(model_name).to(device)
     
@@ -124,25 +124,26 @@ with gr.Blocks() as demo:
             combined_audio = combine_audio_arrays(all_audio)
             all_text += f"**Sentence**: {sentence}\n\n"
             yield (sample_rate, combined_audio), all_text
+
     examples = [
-    [
-        "Once upon a time, in the depth of winter, when the flakes of snow fell like feathers from the clouds, a queen sat sewing at her pal-ace window, which had a carved frame of black wood.",
-        "In an inferior recording quality, a female speaker delivers her slightly expressive and animated words with a fast pace. There's high level of background noise and a very distant-sounding reverberation. Her voice is slightly higher pitched than average.",
-        None,
-    ],
-    [
-        "The Importance of AI Safety.pdf",
-        "Gary's voice is monotone yet slightly fast in delivery, with a very close recording that has no background noise.",
-        None
-    ]            
-]
+        [
+            "Once upon a time, in the depth of winter, when the flakes of snow fell like feathers from the clouds, a queen sat sewing at her palace window, which had a carved frame of black wood.",
+            "In an inferior recording quality, a female speaker delivers her slightly expressive and animated words with a fast pace. There's a high level of background noise and a very distant-sounding reverberation. Her voice is slightly higher pitched than average.",
+            None,
+        ],
+        [
+            "The Importance of AI Safety.pdf",
+            "Gary's voice is monotone yet slightly fast in delivery, with a very close recording that has no background noise.",
+            None
+        ]            
+    ]
 
     input_mode.change(
         fn=lambda choice: [gr.update(visible=choice == "Upload PDF"), gr.update(visible=choice == "Type Text")],
         inputs=input_mode,
         outputs=[pdf_input, text_input],
     )
-    gr.Examples(examples=examples, fn=gen_tts, inputs=inputs, outputs=outputs, cache_examples=True)
+    gr.Examples(examples=examples, fn=run_pipeline, inputs=[input_mode, text_input, description], outputs=[audio_output, markdown_output], cache_examples=True)
     source_lang.change(update_target_lang, inputs=source_lang, outputs=target_lang)
 
     run_button.click(run_pipeline, inputs=[input_mode, pdf_input, text_input, translate_checkbox, source_lang, target_lang, description], outputs=[audio_output, markdown_output])