Update app.py

app.py

@@ -114,13 +114,15 @@ def main(user_image):
   ## identify the face + save the inverted mask
   inverted_mask = identify_face(user_image)
   print("inverted_mask: ", inverted_mask)
-  fp = tempfile.NamedTemporaryFile(mode='wb', suffix=".png") 
-  cv2.imwrite(fp.name, inverted_mask) # save the inverted image mask
-  pil_inverted_mask = PIL.Image.open(fp.name).convert("RGB")
-  print("type(init_image): ", type(init_image))
-  print("type(pil_inverted_mask): ", type(pil_inverted_mask))
-  # sample the new images 
-  return sample_images(init_image, pil_inverted_mask)
+  try: 
+    fp = tempfile.NamedTemporaryFile(mode='wb', suffix=".png") 
+    cv2.imwrite(fp.name, inverted_mask) # save the inverted image mask
+    pil_inverted_mask = PIL.Image.open(fp.name).convert("RGB")
+    # sample the new images 
+    return sample_images(init_image, pil_inverted_mask)
+  else:
+    print("init_image:", type(init_image))
+    return init_image
 
 demo = gr.Interface(main, gr.Image(type="file"), "image")
 demo.launch(debug=True)