Update app.py

app.py

@@ -17,7 +17,7 @@ from src.pix2pix_turbo import Pix2Pix_Turbo
 
 model = Pix2Pix_Turbo("sketch_to_image_stochastic")
 
-ITEMS_NAMES = [ "💡 Lamp","👜 Bag","🛋️ Sofa","🪑 Chair","🏎️ Car","🏍️ Motorbike","🏠 Building", "⌚ Watch"]
+ITEMS_NAMES = [ "💡 Lamp","👜 Bag","🛋️ Sofa","🪑 Chair","🏎️ Car","🏍️ Motorbike","🏠 Building"]
 MAX_SEED = np.iinfo(np.int32).max
 DEFAULT_ITEM_NAME = "💡 Lamp"
 
@@ -44,7 +44,7 @@ def run(image, item_name):
         torch.manual_seed(42)
         B,C,H,W = c_t.shape
         noise = torch.randn((1,4,H//8, W//8), device=c_t.device)
-        output_image = model(c_t, prompt, deterministic=False, r=0.5, noise_map=noise)
+        output_image = model(c_t, prompt, deterministic=False, r=0.4, noise_map=noise)
     output_pil = TF.to_pil_image(output_image[0].cpu()*0.5+0.5)
     input_sketch_uri = pil_image_to_data_uri(Image.fromarray(255-np.array(image)))
     output_image_uri = pil_image_to_data_uri(output_pil)