From 9e678e9a2a01cfa7e589b6c26106683ae3547dc2 Mon Sep 17 00:00:00 2001
From: Qing <cwq1913@gmail.com>
Date: Sun, 26 Mar 2023 21:02:51 +0800
Subject: [PATCH] fix interactiveseg

---
 lama_cleaner/server.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/lama_cleaner/server.py b/lama_cleaner/server.py
index 8ea10b9..d2a20d0 100644
--- a/lama_cleaner/server.py
+++ b/lama_cleaner/server.py
@@ -341,6 +341,13 @@ def run_plugin():
             as_attachment=True,
             attachment_filename=form["filename"],
         )
+    if name == InteractiveSeg.name:
+        return make_response(
+            send_file(
+                io.BytesIO(numpy_to_bytes(bgr_res, "png")),
+                mimetype="image/png",
+            )
+        )
 
     if name == RemoveBG.name:
         rgb_res = cv2.cvtColor(bgr_res, cv2.COLOR_BGRA2RGBA)