darken image

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import torch
 import torch.nn as nn
 import torchvision.transforms as transforms
-from PIL import Image
+from PIL import Image, ImageEnhance
 
 norm_layer = nn.InstanceNorm2d
 
@@ -117,7 +117,10 @@ def predict(input_img, ver):
             drawing = model1(input_img)[0].detach()
 
     drawing = transforms.ToPILImage()(drawing)
-    return drawing
+    enhancer = ImageEnhance.Brightness(drawing)
+    factor = 0.6  # darkens the image
+    im_output = enhancer.enhance(factor)
+    return im_output
 
 
 title = "Image to Line Drawings - Complex and Simple Portraits and Landscapes"