Update app.py

app.py

@@ -36,7 +36,8 @@ def run(image, item_name):
         ones = Image.new("L", (512, 512), 255)
         return ones
     prompt = item_name + " professional 3d model. octane render, highly detailed, volumetric, dramatic lighting"
-    image = image["layers"][0].convert("RGB")
+    inverted_image = PIL.ImageOps.invert(image["layers"][0])
+    image = inverted_image.convert("RGB")
     image_t = TF.to_tensor(image) > 0.5
     with torch.no_grad():
         c_t = image_t.unsqueeze(0).cuda().float()