Update app.py

app.py

@@ -14,6 +14,7 @@ df.columns = ['ISO 639-1', 'ISO 639-2', 'Language Name', 'Native Name']
 df['ISO 639-1'] = df['ISO 639-1'].str.strip()
 
 language_options = [(row['ISO 639-1'], f"{row['ISO 639-1']}") for index, row in df.iterrows()]
+model_size_options = ["tiny", "base", "small", "medium", "large", "large-v2", "large-v3"]  # Add model size options
 
 logging.basicConfig(level=logging.DEBUG)
 
@@ -105,27 +106,28 @@ with gr.Blocks() as blocks_app:
         video_file = gr.Video(label="Upload Video")
         source_language_dropdown = gr.Dropdown(choices=language_options, label="Source Language", value="en")
         target_language_dropdown = gr.Dropdown(choices=language_options, label="Target Language", value="fr")
+        model_size_dropdown = gr.Dropdown(choices=model_size_options, label="Model Size", value="large-v2")  # Model size dropdown
         transcribe_button = gr.Button("Transcribe Video")
         translate_button = gr.Button("Translate Subtitles")
     
     output_video = gr.Video(label="Processed Video")
     output_srt = gr.File(label="Subtitles File (.srt)")
 
-    def transcribe_and_add_subtitles(video_file):
-        transcription = transcribe(video_file, "tiny")
+    def transcribe_and_add_subtitles(video_file, model_size):
+        transcription = transcribe(video_file, model_size)
         srt_path = text_to_srt(transcription)
         output_video_path = add_subtitle_to_video(video_file, srt_path, subtitle_language="eng", soft_subtitle=False)
         return output_video_path, srt_path
     
-    def translate_subtitles_and_add_to_video(video_file, source_language_code, target_language_code):
-        transcription = transcribe(video_file, "tiny")
+    def translate_subtitles_and_add_to_video(video_file, source_language_code, target_language_code, model_size):
+        transcription = transcribe(video_file, model_size)
         srt_path = text_to_srt(transcription)
         translated_srt_path = translate_srt(srt_path, source_language_code, target_language_code)
         output_video_path = add_subtitle_to_video(video_file, translated_srt_path, target_language_code, soft_subtitle=False)
         return output_video_path, translated_srt_path
     
-    transcribe_button.click(transcribe_and_add_subtitles, inputs=video_file, outputs=[output_video, output_srt])
-    translate_button.click(translate_subtitles_and_add_to_video, inputs=[video_file, source_language_dropdown, target_language_dropdown], outputs=[output_video, output_srt])
+    transcribe_button.click(transcribe_and_add_subtitles, inputs=[video_file, model_size_dropdown], outputs=[output_video, output_srt])
+    translate_button.click(translate_subtitles_and_add_to_video, inputs=[video_file, source_language_dropdown, target_language_dropdown, model_size_dropdown], outputs=[output_video, output_srt])
 
 # Lancement de l'application
 blocks_app.launch()