From d39c0ca01717dd0f7f664867c2cd8c6b1a64c76d Mon Sep 17 00:00:00 2001
From: Roger Condori <114810545+R3gm@users.noreply.github.com>
Date: Sat, 14 Oct 2023 07:26:24 -0400
Subject: [PATCH] fixed error import in inpaint

---
 stablepy/diffusers_vanilla/model.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/stablepy/diffusers_vanilla/model.py b/stablepy/diffusers_vanilla/model.py
index bc0c65c..29a63d2 100644
--- a/stablepy/diffusers_vanilla/model.py
+++ b/stablepy/diffusers_vanilla/model.py
@@ -1390,7 +1390,9 @@ def __call__(
                 numpy_array_mask = np.array(mask_control_img, dtype=np.uint8)
                 array_rgb_mask = numpy_array_mask[:, :, :3]
             elif not gui_active:
-                # Convert control image for draw
+                # Convert control image to draw
+                import base64
+                import matplotlib.pyplot as plt
                 name_without_extension = os.path.splitext(image.split("/")[-1])[0]
                 image64 = base64.b64encode(open(image, "rb").read())
                 image64 = image64.decode("utf-8")