Update app.py

app.py

@@ -117,7 +117,7 @@ def gpu_predict(model_path, device_map, torch_dtype,
             assert output_res.count("think>") == 2
             new_prompt_ret = output_res.split("think>")[-1]
             if new_prompt_ret.startswith("\n"):
-                new_prompt = new_prompt[1:]
+                new_prompt = new_prompt_ret[1:]
             new_prompt = replace_single_quotes(new_prompt)
         except Exception:
             # 如果格式不符合预期，则直接回退为原始输入