Update app.py

app.py

@@ -32,11 +32,11 @@ def pil_image_to_data_uri(img, format='PNG'):
 def run(image, item_name):
     print("sketch updated")
     print(image)
-    if image["composite"] is None:
+    if image["layers"][0] is None:
         ones = Image.new("L", (512, 512), 255)
         return ones
     prompt = item_name + " professional 3d model. octane render, highly detailed, volumetric, dramatic lighting"
-    image = image["composite"].convert("RGB")
+    image = image["layers"][0].convert("RGB")
     image_t = TF.to_tensor(image) > 0.5
     with torch.no_grad():
         c_t = image_t.unsqueeze(0).cuda().float()