Spaces:
Runtime error
Runtime error
senior-sigan
commited on
Commit
•
284b74f
1
Parent(s):
7e65230
remove progress bar because of queue error
Browse files
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
|
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)
|