Spaces:
Sleeping
Sleeping
Avatar post processing
Browse files
app.py
CHANGED
|
@@ -154,7 +154,7 @@ def create_avatar(img: Image.Image, prompt: str, strength: float, guidance_scale
|
|
| 154 |
with torch.autocast("cuda"):
|
| 155 |
result = sd_pipe(prompt=prompt, image=img_resized, strength=strength, guidance_scale=guidance_scale)
|
| 156 |
|
| 157 |
-
avatar = result.images[0]
|
| 158 |
|
| 159 |
return avatar
|
| 160 |
|
|
|
|
| 154 |
with torch.autocast("cuda"):
|
| 155 |
result = sd_pipe(prompt=prompt, image=img_resized, strength=strength, guidance_scale=guidance_scale)
|
| 156 |
|
| 157 |
+
avatar = enhance_face(result.images[0])
|
| 158 |
|
| 159 |
return avatar
|
| 160 |
|