Просмотр исходного кода

Merge pull request #288 from charles7668/fix/thread-value-error

Fix an issue where translation would fail if the threads value was not provided.
Byaidu 1 год назад
Родитель
Сommit
cf46f09f5c
1 измененных файлов с 9 добавлено и 2 удалено
  1. 9 2
      pdf2zh/gui.py

+ 9 - 2
pdf2zh/gui.py

@@ -188,6 +188,11 @@ def translate_file(
     def progress_bar(t: tqdm.tqdm):
     def progress_bar(t: tqdm.tqdm):
         progress(t.n / t.total, desc="Translating...")
         progress(t.n / t.total, desc="Translating...")
 
 
+    try:
+        threads = int(threads)
+    except ValueError:
+        threads = 1
+
     param = {
     param = {
         "files": [str(file_raw)],
         "files": [str(file_raw)],
         "pages": selected_page,
         "pages": selected_page,
@@ -195,7 +200,7 @@ def translate_file(
         "lang_out": lang_to,
         "lang_out": lang_to,
         "service": f"{translator.name}",
         "service": f"{translator.name}",
         "output": output,
         "output": output,
-        "thread": int(threads),
+        "thread": threads,
         "callback": progress_bar,
         "callback": progress_bar,
         "cancellation_event": cancellation_event_map[session_id],
         "cancellation_event": cancellation_event_map[session_id],
         "envs": _envs,
         "envs": _envs,
@@ -342,7 +347,9 @@ with gr.Blocks(
 
 
             with gr.Accordion("Open for More Experimental Options!", open=False):
             with gr.Accordion("Open for More Experimental Options!", open=False):
                 gr.Markdown("#### Experimental")
                 gr.Markdown("#### Experimental")
-                threads = gr.Textbox(label="number of threads", interactive=True)
+                threads = gr.Textbox(
+                    label="number of threads", interactive=True, value="1"
+                )
                 prompt = gr.Textbox(
                 prompt = gr.Textbox(
                     label="Custom Prompt for llm", interactive=True, visible=False
                     label="Custom Prompt for llm", interactive=True, visible=False
                 )
                 )