update

app.py

@@ -1,4 +1,7 @@
+import nltk
 import subprocess
 
-# 运行 src/main.py 文件
+nltk.download('punkt')
+nltk.download('punkt_tab')
+
 subprocess.run(["python", "src/main.py"])

src/main.py

@@ -73,8 +73,9 @@ def create_interface():
 
         with gr.Row():
             with gr.Column():
-                model_gr = gr.Dropdown(choices=["gpt-3.5-turbo", "gpt-4o", "gpt-4o-mini"], label="🤖 Select your Model")
+                model_gr = gr.Dropdown(choices=["deepseek-chat", "deepseek-reasoner", "gpt-3.5-turbo", "gpt-4o", "gpt-4o-mini"], label="🤖 Select your Model")
                 api_key_gr = gr.Textbox(label="🔑 Enter your API-Key")
+                base_url_gr = gr.Textbox(label="🔗 Enter your Base-URL", value="DEFAULT")
             with gr.Column():
                 task_gr = gr.Dropdown(choices=["Base", "NER", "RE", "EE"], label="🎯 Select your Task")
                 use_file_gr = gr.Checkbox(label="📂 Use File", value=True)
@@ -112,10 +113,19 @@ def create_interface():
                 gr.update(value=example["constraint"], visible=example["task"] in ["NER", "RE", "EE"]),
             )
 
-        def submit(model, api_key, task, instruction, constraint, text, use_file, file_path):
+        def submit(model, api_key, base_url, task, instruction, constraint, text, use_file, file_path):
             try:
-                # 创建 Pipeline 实例
-                pipeline = Pipeline(ChatGPT(model_name_or_path=model, api_key=api_key))
+                if base_url == "DEFAULT":
+                    if model in ["gpt-3.5-turbo", "gpt-4o", "gpt-4o-mini"]:
+                        pipeline = Pipeline(ChatGPT(model_name_or_path=model, api_key=api_key))
+                    elif model in ["deepseek-chat", "deepseek-reasoner"]:
+                        pipeline = Pipeline(DeepSeek(model_name_or_path=model, api_key=api_key))
+                else:
+                    if model in ["gpt-3.5-turbo", "gpt-4o", "gpt-4o-mini"]:
+                        pipeline = Pipeline(ChatGPT(model_name_or_path=model, api_key=api_key, base_url=base_url))
+                    elif model in ["deepseek-chat", "deepseek-reasoner"]:
+                        pipeline = Pipeline(DeepSeek(model_name_or_path=model, api_key=api_key, base_url=base_url))
+
                 if task == "Base":
                     instruction = instruction
                     constraint = ""
@@ -199,6 +209,7 @@ def create_interface():
             inputs=[
                 model_gr,
                 api_key_gr,
+                base_url_gr,
                 task_gr,
                 instruction_gr,
                 constraint_gr,
@@ -214,6 +225,7 @@ def create_interface():
             outputs=[
                 model_gr,
                 api_key_gr,
+                base_url_gr,
                 task_gr,
                 instruction_gr,
                 constraint_gr,