Update 01_🎥_Input_YouTube_Link.py

01_🎥_Input_YouTube_Link.py

@@ -218,11 +218,8 @@ def main():
                 st.download_button(label="Download Subtitled Video",
                                     data=video_with_subs,
                                     file_name=f"{title} with subtitles.mp4")
-    elif task == "Translate":
-        default_language = "French"
-        target = st.selectbox("Select Language", list(languages.keys()), index=list(languages.keys()).index(default_language))
-        target_code = languages[selected_language]
-        if st.button("Translate"):
+    elif task == "Translate with Whisper":
+        if st.button("Translate to English"):
             author, title, description, thumbnail, length, views = populate_metadata(link)
             results = inference(link, loaded_model, task)
             video = download_video(link)
@@ -279,6 +276,11 @@ def main():
                 st.download_button(label="Download Subtitled Video ",
                                     data=video_with_subs,
                                     file_name=f"{title} with subtitles.mp4")
+    elif task == "Translate with Sema":
+        default_language = "French"
+        target = st.selectbox("Select Language", list(flores_codes.keys()), index=list(flores_codes.keys()).index(default_language))
+        target_code = flores_codes[target]
+        
     else:
         st.error("Please select a task.")