Update app.py

app.py

@@ -20,7 +20,10 @@ model = Pix2Pix_Turbo("sketch_to_image_stochastic")
 ITEMS_NAMES = [ "💡 Lamp","👜 Bag","🛋️ Sofa","🪑 Chair","🏎️ Car","🏍️ Motorbike"]
 MAX_SEED = np.iinfo(np.int32).max
 DEFAULT_ITEM_NAME = "💡 Lamp"
-def empty_input_image(): 
+COLORS = ["blue", "red", "black", "yellow", "green", "pink", "grey"]
+SELECTED_COLOR = random.choice(COLORS)
+def empty_input_image():
+    SELECTED_COLOR = random.choice(COLORS)
     return { 'background': Image.new("L", (512, 512), 255),
              'layers': [Image.new("L", (512, 512), 255),Image.new("L", (512, 512), 255)],
              'composite': Image.new("L", (512, 512), 255)}
@@ -42,7 +45,7 @@ def run(image, item_name):
         ones = empty_image
         return ones
     print(item_name.split()[1])
-    prompt = item_name.split()[1] + " professional 3d model. octane render, highly detailed, volumetric, dramatic lighting"
+    prompt = SELECTED_COLOR + " " + 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