Update app.py

app.py

@@ -38,7 +38,7 @@ def run(image, item_name):
     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"
+    prompt = item_name.split()[1] + " professional 3d model. octane render, highly detailed, volumetric, dramatic lighting"
     inverted_image = ImageOps.invert(image["composite"])
     converted_image = inverted_image.convert("RGB")
     image_t = TF.to_tensor(converted_image) > 0.5