Spaces:
Sleeping
Sleeping
File size: 2,590 Bytes
100f3e3 0b349b6 6e54822 4c2e294 100f3e3 d0e5d23 0b349b6 acd3a4d 0b349b6 100f3e3 5a4c72f 02be53b acd3a4d 5a4c72f 100f3e3 0b349b6 5a4c72f 0b349b6 5c92fe7 0b349b6 100f3e3 acd3a4d 100f3e3 acd3a4d 100f3e3 5c92fe7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
import gradio as gr
from src.translate_any_doc import translate_document
from src.salamandraTA7b_translator import SalamandraTA7bTranslator
from src.aligner import Aligner
import os
config_folder = 'fast_align_config'
temp_folder = 'tmp'
hf_token = os.environ.get('HF_TOKEN')
if hf_token:
print("Read HF token")
else:
print("HF token not found")
translator = SalamandraTA7bTranslator(hf_token)
def upload_file(filepath, source_lang, target_lang):
aligner = Aligner(config_folder, source_lang, target_lang, temp_folder)
for status, translated_file_name in translate_document(filepath, source_lang, target_lang, translator, aligner):
if translated_file_name: # finished
yield [gr.UploadButton(visible=False),
gr.DownloadButton(label=f"Download {translated_file_name}", value=translated_file_name,
visible=True, interactive=True),
gr.Textbox(visible=False)]
else:
yield [gr.UploadButton(visible=False),
gr.DownloadButton(visible=False),
gr.Textbox(value=status, visible=True)]
def before_processing():
return [
gr.UploadButton(visible=False),
gr.Textbox(value="Processing...", visible=True),
]
def download_file():
return [gr.UploadButton(visible=True), gr.DownloadButton(visible=False)]
with gr.Blocks() as demo:
with gr.Tab("Text"):
gr.Interface(fn=translator.translate, inputs=["text", "text", "text"], outputs="text")
with gr.Tab("Documents"):
with gr.Row():
dropdown1 = gr.Dropdown(label="Source language", choices=["en", "ca"], value=None, interactive=True)
dropdown2 = gr.Dropdown(label="Target language", choices=["en", "ca"], value=None, interactive=True)
gr.Markdown("First upload a file and and then you'll be able download it (but only once!)")
with gr.Row():
u = gr.UploadButton("Upload a file", file_count="single")
d = gr.DownloadButton("Download the file", visible=False)
status_text = gr.Textbox(label="Status", visible=False)
u.upload(fn=before_processing, inputs=None, outputs=[u, status_text]).then(upload_file,
[u, dropdown1, dropdown2],
[u, d, status_text])
d.click(download_file, None, [u, d])
if __name__ == "__main__":
demo.launch(server_name="0.0.0.0", server_port=7860)
|