Browse Source

增加http接口调用

zhy 3 weeks ago
parent
commit
c9706040e1
1 changed files with 8 additions and 4 deletions
  1. 8 4
      watermark_verify/app.py

+ 8 - 4
watermark_verify/app.py

@@ -1,4 +1,4 @@
-from flask import Flask, jsonify
+from flask import Flask, jsonify, request
 
 from watermark_verify.exceptions import BusinessException
 from watermark_verify import verify_tool
@@ -13,9 +13,13 @@ def create_app():
         return jsonify({"message": ex.message, 'code': ex.code}), 500
 
     @app.route("/decrypt_model", methods=['POST'])
-    def detect_watermark():
+    def decrypt_model():
         # TODO 根据工标需要进行HTTP接口开发
-        verify_tool.label_verification(model_filename='')
-        return "Hello World!"
+        data = request.json
+        model_filename = data.get('model_filename') # 模型权重文件位置
+
+        result = verify_tool.label_verification(model_filename=model_filename)
+        print(f"模型水印检测结果: {result}")
+        return jsonify({"result": result})
 
     return app