ソースを参照

fix : prompt from gui error.

hellofinch 1 年間 前
コミット
6639e9eda4
1 ファイル変更2 行追加1 行削除
  1. 2 1
      pdf2zh/gui.py

+ 2 - 1
pdf2zh/gui.py

@@ -10,6 +10,7 @@ import gradio as gr
 import requests
 import tqdm
 from gradio_pdf import PDF
+from string import Template
 
 from pdf2zh import __version__
 from pdf2zh.high_level import translate
@@ -276,7 +277,7 @@ def translate_file(
         "callback": progress_bar,
         "cancellation_event": cancellation_event_map[session_id],
         "envs": _envs,
-        "prompt": prompt,
+        "prompt": Template(prompt),
         "model": ModelInstance.value,
     }
     try: