|
@@ -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})
|
|
|
|