senior-sigan commited on
Commit
284b74f
1 Parent(s): 7e65230

remove progress bar because of queue error

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -36,13 +36,11 @@ def join_image_h(im1: PIL.Image.Image, im2: PIL.Image.Image) -> PIL.Image.Image:
36
  def predict(
37
  image: PIL.Image.Image,
38
  style: str,
39
- progress = gr.Progress(),
40
  *,
41
  face_aligner: FaceAligner,
42
  encoder: Encoder,
43
  generator: Dict[str, Generator],
44
  ) -> Tuple[List[PIL.Image.Image], Optional[str]]:
45
- progress(0, desc="Starting...")
46
  images = face_aligner.align(image)
47
  if len(images) == 0:
48
  error_msg = "Cannot find any face in photo"
@@ -50,7 +48,7 @@ def predict(
50
  return [PIL.Image.new('RGB', (1, 1))], error_msg
51
 
52
  results = []
53
- for img in progress.tqdm(images):
54
  x = encoder.predict(img)
55
  gen_img = generator[style].predict(x)
56
  result = join_image_h(img, gen_img)
 
36
  def predict(
37
  image: PIL.Image.Image,
38
  style: str,
 
39
  *,
40
  face_aligner: FaceAligner,
41
  encoder: Encoder,
42
  generator: Dict[str, Generator],
43
  ) -> Tuple[List[PIL.Image.Image], Optional[str]]:
 
44
  images = face_aligner.align(image)
45
  if len(images) == 0:
46
  error_msg = "Cannot find any face in photo"
 
48
  return [PIL.Image.new('RGB', (1, 1))], error_msg
49
 
50
  results = []
51
+ for img in images:
52
  x = encoder.predict(img)
53
  gen_img = generator[style].predict(x)
54
  result = join_image_h(img, gen_img)