bug fixed for 2 color image

app.py

@@ -55,6 +55,10 @@ def create_meme(prompt):
     
     img_path = meme_filename[closest_text_idx]
     img = mpimg.imread(f'kaggle/input/memedata/{img_path}')
+    if len(img.shape) == 2:
+        img = np.stack((img,)*3, axis=-1)
+
+    img_height, img_width = img.shape[:2]
     
     white_space_height = 100
     white_space = np.ones((white_space_height, img.shape[1], img.shape[2]), dtype=img.dtype) * 255