Add translation
Browse files
app.py
CHANGED
@@ -22,10 +22,10 @@ CONCATENATE_TRANSCRIPTIONS = True
|
|
22 |
TRANSLATE_TRANSCRIPTIONS = True
|
23 |
ADD_SUBTITLES_TO_VIDEO = True
|
24 |
REMOVE_FILES = False
|
25 |
-
if
|
26 |
-
SECONDS = 150
|
27 |
-
else:
|
28 |
SECONDS = 300
|
|
|
|
|
29 |
|
30 |
YOUTUBE = "youtube"
|
31 |
TWITCH = "twitch"
|
@@ -458,7 +458,7 @@ def subtify():
|
|
458 |
with gr.Blocks() as demo:
|
459 |
# Layout
|
460 |
gr.Markdown("""# Subtify""")
|
461 |
-
gr.Markdown(f"
|
462 |
# model = transformers.AutoModel.from_pretrained("huggingface/my_model")
|
463 |
# gr.Markdown(f"model.config.url: {model.config.url}")
|
464 |
token = os.getenv("HF_TOKEN")
|
@@ -524,7 +524,7 @@ def subtify():
|
|
524 |
)
|
525 |
subtify_button.click(fn=get_audio_and_video_from_video, inputs=[url_textbox, stream_page], outputs=[original_audio, original_audio_path, original_video_path])
|
526 |
original_audio.change(fn=trascribe_audio, inputs=[original_audio_path, source_languaje], outputs=[original_audio_transcribed, original_audio_transcribed_path])
|
527 |
-
|
528 |
# original_audio_translated.change(fn=add_translated_subtitles_to_video, inputs=[original_video_path, original_audio_path, original_audio_translated_path], outputs=subtitled_video)
|
529 |
|
530 |
demo.launch()
|
|
|
22 |
TRANSLATE_TRANSCRIPTIONS = True
|
23 |
ADD_SUBTITLES_TO_VIDEO = True
|
24 |
REMOVE_FILES = False
|
25 |
+
if DEVICE == "cpu":
|
|
|
|
|
26 |
SECONDS = 300
|
27 |
+
else:
|
28 |
+
SECONDS = 50
|
29 |
|
30 |
YOUTUBE = "youtube"
|
31 |
TWITCH = "twitch"
|
|
|
458 |
with gr.Blocks() as demo:
|
459 |
# Layout
|
460 |
gr.Markdown("""# Subtify""")
|
461 |
+
gr.Markdown(f"translate, Python: {sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro}")
|
462 |
# model = transformers.AutoModel.from_pretrained("huggingface/my_model")
|
463 |
# gr.Markdown(f"model.config.url: {model.config.url}")
|
464 |
token = os.getenv("HF_TOKEN")
|
|
|
524 |
)
|
525 |
subtify_button.click(fn=get_audio_and_video_from_video, inputs=[url_textbox, stream_page], outputs=[original_audio, original_audio_path, original_video_path])
|
526 |
original_audio.change(fn=trascribe_audio, inputs=[original_audio_path, source_languaje], outputs=[original_audio_transcribed, original_audio_transcribed_path])
|
527 |
+
original_audio_transcribed.change(fn=translate_transcription, inputs=[original_audio_transcribed_path, source_languaje, target_languaje], outputs=[original_audio_translated, original_audio_translated_path])
|
528 |
# original_audio_translated.change(fn=add_translated_subtitles_to_video, inputs=[original_video_path, original_audio_path, original_audio_translated_path], outputs=subtitled_video)
|
529 |
|
530 |
demo.launch()
|