Update run/gradio_ootd.py

run/gradio_ootd.py

@@ -1,7 +1,7 @@
-from flask import Flask, request, jsonify
+from flask import Flask, request, jsonify , send_file
 import torch
 from PIL import Image, ImageOps
-
+import base64
 from utils_ootd import get_mask_location
 from preprocess.openpose.run_openpose import OpenPose
 from preprocess.humanparsing.run_parsing import Parsing
@@ -69,7 +69,14 @@ def process_hd():
             seed=seed,
         )
 
-    return jsonify(result=images)
+        base64_images = []
+        for img in images:
+            buffered = BytesIO()
+            img.save(buffered, format="JPEG")
+            img_str = base64.b64encode(buffered.getvalue()).decode('utf-8')
+            base64_images.append(img_str)
+
+    return jsonify(images=base64_images)
 
 @app.route("/process_dc", methods=["POST"])
 def process_dc():
@@ -120,8 +127,14 @@ def process_dc():
             image_scale=image_scale,
             seed=seed,
         )
-
-    return jsonify(result=images)
+        base64_images = []
+        for img in images:
+            buffered = BytesIO()
+            img.save(buffered, format="JPEG")
+            img_str = base64.b64encode(buffered.getvalue()).decode('utf-8')
+            base64_images.append(img_str)
+
+    return jsonify(images=base64_images)
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)