Browse Source

修改引入文件

zhy 2 weeks ago
parent
commit
959231bdd0
1 changed files with 2 additions and 2 deletions
  1. 2 2
      watermark_verify/app.py

+ 2 - 2
watermark_verify/app.py

@@ -1,7 +1,7 @@
 from flask import Flask, jsonify, request
 
 from watermark_verify.exceptions import BusinessException
-from watermark_verify import verify_tool
+from watermark_verify import verify_tool_mix
 
 
 def create_app():
@@ -21,7 +21,7 @@ def create_app():
         mode = data.get('mode')
         model_type = data.get('model_type')
 
-        result = verify_tool.label_verification(model_filename=model_filename, framework=framework, mode=mode, model_type=model_type)
+        result = verify_tool_mix.label_verification(model_filename=model_filename, framework=framework, mode=mode, model_type=model_type)
         print(f"模型水印检测结果: {result}")
         return jsonify({"result": result})