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 is None:
+    if image["composite"] 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.convert("RGB")
+    image = image["composite"].convert("RGB")
     image_t = TF.to_tensor(image) > 0.5
     with torch.no_grad():
         c_t = image_t.unsqueeze(0).cuda().float()
@@ -187,7 +187,7 @@ with gr.Blocks(css="style.css") as demo:
         item = gr.Radio(choices=ITEMS_NAMES, value=DEFAULT_ITEM_NAME, show_label=False, container=False)
 
     demo.load(None,None,None,js=scripts)
-    inputs = [image["composite"], item]
+    inputs = [image, item]
     outputs = [result]
     item.change(fn=run, inputs=inputs, outputs=outputs)
     image.change(fn=run, inputs=inputs, outputs=outputs, trigger_mode="always_last")