Update app.py

app.py

@@ -133,7 +133,7 @@ def run(sketch, caption, threshold, seed):
             hsv_image[..., 2][mask] = 0  # Value set to 0, making it black
     
     mask_tensor_org = sketch2[:,:,0]/255
-    hsv_image[mask_tensor_org==1] = [0,0,1]
+    hsv_image[mask_tensor_org>=0.5] = [0,0,1]
 
     # Convert the HSV image back to RGB to display and save
     rgb_image = hsv_to_rgb(hsv_image)