FlowKal commited on
Commit
464fb6c
·
verified ·
1 Parent(s): 1ba882e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  import torch
3
  import pickle
4
- from PIL import Image
5
  import numpy as np
6
  import cv2
7
  import torchvision.transforms as transforms
@@ -62,7 +62,8 @@ def preprocess_single_image(pil_image: Image.Image) -> torch.Tensor:
62
  """
63
  print(pil_image)
64
  img_gray = pil_image['composite'].convert('L')
65
-
 
66
  # 2. Конвертируем в numpy для трансформаций из cv2
67
  img_np = np.array(img_gray)
68
 
 
1
  import gradio as gr
2
  import torch
3
  import pickle
4
+ from PIL import Image,ImageOps
5
  import numpy as np
6
  import cv2
7
  import torchvision.transforms as transforms
 
62
  """
63
  print(pil_image)
64
  img_gray = pil_image['composite'].convert('L')
65
+
66
+ img_gray = ImageOps.invert(img_gray)
67
  # 2. Конвертируем в numpy для трансформаций из cv2
68
  img_np = np.array(img_gray)
69