Spaces:
Sleeping
Sleeping
added textbox to use user token, remove text translation for now
Browse files- gradio_app.py +10 -12
gradio_app.py
CHANGED
@@ -6,17 +6,13 @@ import os
|
|
6 |
|
7 |
config_folder = 'fast_align_config'
|
8 |
temp_folder = 'tmp'
|
9 |
-
hf_token = os.environ.get('HF_TOKEN')
|
10 |
|
11 |
-
if hf_token:
|
12 |
-
print("Read HF token")
|
13 |
-
else:
|
14 |
-
print("HF token not found")
|
15 |
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
|
|
|
20 |
aligner = Aligner(config_folder, source_lang, target_lang, temp_folder)
|
21 |
for status, translated_file_name in translate_document(filepath, source_lang, target_lang, translator, aligner):
|
22 |
if translated_file_name: # finished
|
@@ -42,12 +38,13 @@ def download_file():
|
|
42 |
|
43 |
|
44 |
with gr.Blocks() as demo:
|
45 |
-
with gr.Tab("Text"):
|
46 |
-
|
47 |
with gr.Tab("Documents"):
|
48 |
with gr.Row():
|
49 |
dropdown1 = gr.Dropdown(label="Source language", choices=["en", "ca"], value=None, interactive=True)
|
50 |
dropdown2 = gr.Dropdown(label="Target language", choices=["en", "ca"], value=None, interactive=True)
|
|
|
51 |
gr.Markdown("First upload a file and and then you'll be able download it (but only once!)")
|
52 |
with gr.Row():
|
53 |
u = gr.UploadButton("Upload a file", file_count="single")
|
@@ -55,7 +52,8 @@ with gr.Blocks() as demo:
|
|
55 |
status_text = gr.Textbox(label="Status", visible=False)
|
56 |
|
57 |
u.upload(fn=before_processing, inputs=None, outputs=[u, status_text]).then(upload_file,
|
58 |
-
[u, dropdown1, dropdown2
|
|
|
59 |
[u, d, status_text])
|
60 |
d.click(download_file, None, [u, d])
|
61 |
if __name__ == "__main__":
|
|
|
6 |
|
7 |
config_folder = 'fast_align_config'
|
8 |
temp_folder = 'tmp'
|
|
|
9 |
|
|
|
|
|
|
|
|
|
10 |
|
11 |
+
def upload_file(filepath, source_lang, target_lang, user_token):
|
12 |
+
hf_token = os.environ.get('HF_TOKEN')
|
13 |
+
if user_token:
|
14 |
+
hf_token = user_token
|
15 |
+
translator = SalamandraTA7bTranslator(hf_token)
|
16 |
aligner = Aligner(config_folder, source_lang, target_lang, temp_folder)
|
17 |
for status, translated_file_name in translate_document(filepath, source_lang, target_lang, translator, aligner):
|
18 |
if translated_file_name: # finished
|
|
|
38 |
|
39 |
|
40 |
with gr.Blocks() as demo:
|
41 |
+
# with gr.Tab("Text"):
|
42 |
+
# gr.Interface(fn=translator.translate, inputs=["text", "text", "text"], outputs="text")
|
43 |
with gr.Tab("Documents"):
|
44 |
with gr.Row():
|
45 |
dropdown1 = gr.Dropdown(label="Source language", choices=["en", "ca"], value=None, interactive=True)
|
46 |
dropdown2 = gr.Dropdown(label="Target language", choices=["en", "ca"], value=None, interactive=True)
|
47 |
+
token_textbox = gr.Textbox(label="Huggingface token (optional)", visible=True)
|
48 |
gr.Markdown("First upload a file and and then you'll be able download it (but only once!)")
|
49 |
with gr.Row():
|
50 |
u = gr.UploadButton("Upload a file", file_count="single")
|
|
|
52 |
status_text = gr.Textbox(label="Status", visible=False)
|
53 |
|
54 |
u.upload(fn=before_processing, inputs=None, outputs=[u, status_text]).then(upload_file,
|
55 |
+
[u, dropdown1, dropdown2,
|
56 |
+
token_textbox],
|
57 |
[u, d, status_text])
|
58 |
d.click(download_file, None, [u, d])
|
59 |
if __name__ == "__main__":
|