ohayonguy commited on
Commit
0305973
1 Parent(s): ad47941

improved description

Browse files
Files changed (1) hide show
  1. app.py +0 -4
app.py CHANGED
@@ -3,7 +3,6 @@ if os.getenv('SPACES_ZERO_GPU') == "true":
3
  os.environ['SPACES_ZERO_GPU'] = "1"
4
  os.environ['K_DIFFUSION_USE_COMPILE'] = "0"
5
  import spaces
6
- from facelib.utils.misc import is_gray
7
  import cv2
8
  import gradio as gr
9
  import torch
@@ -65,9 +64,6 @@ def enhance_face(img, face_helper, has_aligned, num_flow_steps, only_center_face
65
  if has_aligned:
66
  # the input faces are already cropped and aligned
67
  img = cv2.resize(img, (512, 512), interpolation=cv2.INTER_LINEAR)
68
- face_helper.is_gray = is_gray(img, threshold=5)
69
- if face_helper.is_gray:
70
- print('\tgrayscale input: True')
71
  face_helper.cropped_faces = [img]
72
  else:
73
  face_helper.read_image(img)
 
3
  os.environ['SPACES_ZERO_GPU'] = "1"
4
  os.environ['K_DIFFUSION_USE_COMPILE'] = "0"
5
  import spaces
 
6
  import cv2
7
  import gradio as gr
8
  import torch
 
64
  if has_aligned:
65
  # the input faces are already cropped and aligned
66
  img = cv2.resize(img, (512, 512), interpolation=cv2.INTER_LINEAR)
 
 
 
67
  face_helper.cropped_faces = [img]
68
  else:
69
  face_helper.read_image(img)