Update flux1_img2img.py

flux1_img2img.py

@@ -10,9 +10,8 @@ import spaces
 @spaces.GPU
 def process_image(image,mask_image,prompt="a person",model_id="black-forest-labs/FLUX.1-schnell",strength=0.75,seed=0,num_inference_steps=4):
     print("start process image process_image")
-    if image == None:
-        print("empty input image returned")
-        return None
+    if image is None or not hasattr(image, 'size'):
+        raise gr.Error("Please upload an image.")
     
     pipe = FluxImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16)
     pipe.to("cuda")