Update app.py

app.py

@@ -113,16 +113,15 @@ def main(user_image):
   init_image.save(user_image.name) # save the resized image
   ## identify the face + save the inverted mask
   inverted_mask = identify_face(user_image)
-  print("inverted_mask: ", 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:
+  if inverted_mask == None:
     print("init_image:", type(init_image))
     return init_image
+  print("inverted_mask: ", inverted_mask)
+  fp = tempfile.NamedTemporaryFile(mode='wb', suffix=".png") 
+  cv2.imwrite(fp.name, inverted_mask) # save the inverted image 
+  pil_inverted_mask = PIL.Image.open(fp.name).convert("RGB")
+  # sample the new 
+  return sample_images(init_image, pil_inverted_mask)
 
 demo = gr.Interface(main, gr.Image(type="file"), "image")
 demo.launch(debug=True)