Update app.py

app.py

@@ -113,16 +113,17 @@ def gpu_predict(model_path, device_map, torch_dtype,
         )
         output_res = output_text[0]
         # 兼容原逻辑：提取 think> 之后的内容
-        try:
-            assert output_res.count("think>") == 2
-            new_prompt_ret = output_res.split("think>")[-1]
-            if new_prompt_ret.startswith("\n"):
-                new_prompt = new_prompt_ret[1:]
-            new_prompt = replace_single_quotes(new_prompt)
-        except Exception:
-            # 如果格式不符合预期，则直接回退为原始输入
-            new_prompt_ret = org_prompt_cot
-        return new_prompt_ret, ""
+        # try:
+        #     assert output_res.count("think>") == 2
+        #     new_prompt_ret = output_res.split("think>")[-1]
+        #     if new_prompt_ret.startswith("\n"):
+        #         new_prompt = new_prompt_ret[1:]
+        #     new_prompt = replace_single_quotes(new_prompt)
+        # except Exception:
+        #     # 如果格式不符合预期，则直接回退为原始输入
+        #     new_prompt_ret = org_prompt_cot
+        # return new_prompt_ret, ""
+        return output_res, ""
     except Exception as e:
         # 失败则返回原始提示词和错误信息
         return org_prompt_cot, f"推理失败：{e}"