Update pred_color.py

pred_color.py

@@ -94,6 +94,8 @@ import os
 def single_pred_features(image):
     if type(image) == type("") and os.path.exists(image):
         image = Image.open(image).convert("RGB")
+    elif hasattr(image, "shape"):
+        image = Image.fromarray(image).convert("RGB")
     else:
         image = image.convert("RGB")
     cv2_image = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)