xinjie.wang commited on
Commit
1f34dd4
·
1 Parent(s): 7b331b8
Files changed (1) hide show
  1. common.py +3 -0
common.py CHANGED
@@ -210,6 +210,9 @@ def preprocess_image_fn(
210
  image: str | np.ndarray | Image.Image,
211
  req: gr.Request,
212
  ) -> Image.Image:
 
 
 
213
  if isinstance(image, str):
214
  image = Image.open(image)
215
  elif isinstance(image, np.ndarray):
 
210
  image: str | np.ndarray | Image.Image,
211
  req: gr.Request,
212
  ) -> Image.Image:
213
+ if image is None:
214
+ return
215
+
216
  if isinstance(image, str):
217
  image = Image.open(image)
218
  elif isinstance(image, np.ndarray):