mjuvilla commited on
Commit
85b75e6
·
1 Parent(s): d0e5d23

added textbox to use user token, remove text translation for now

Browse files
Files changed (1) hide show
  1. 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
- translator = SalamandraTA7bTranslator(hf_token)
17
-
18
-
19
- def upload_file(filepath, source_lang, target_lang):
 
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
- gr.Interface(fn=translator.translate, inputs=["text", "text", "text"], outputs="text")
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__":