Spaces:
Running
on
Zero
Running
on
Zero
Rishi Desai
commited on
Commit
·
edd3675
1
Parent(s):
656b972
temp removing spaces
Browse files- face_enhance.py +2 -2
face_enhance.py
CHANGED
|
@@ -3,7 +3,7 @@ import random
|
|
| 3 |
import sys
|
| 4 |
from typing import Sequence, Mapping, Any, Union
|
| 5 |
import torch
|
| 6 |
-
import spaces
|
| 7 |
COMFYUI_PATH = "./ComfyUI"
|
| 8 |
|
| 9 |
"""
|
|
@@ -313,7 +313,7 @@ def save_comfy_images(images, output_dirs):
|
|
| 313 |
pil_image = Image.fromarray(numpy_image)
|
| 314 |
pil_image.save(output_dirs[idx])
|
| 315 |
|
| 316 |
-
@spaces.GPU
|
| 317 |
def face_enhance(face_image: str, input_image: str, output_image: str, dist_image: str = None, positive_prompt: str = "", id_weight: float = 0.75):
|
| 318 |
initialize_models() # Ensure models are loaded
|
| 319 |
main(face_image, input_image, output_image, dist_image, positive_prompt, id_weight)
|
|
|
|
| 3 |
import sys
|
| 4 |
from typing import Sequence, Mapping, Any, Union
|
| 5 |
import torch
|
| 6 |
+
# import spaces
|
| 7 |
COMFYUI_PATH = "./ComfyUI"
|
| 8 |
|
| 9 |
"""
|
|
|
|
| 313 |
pil_image = Image.fromarray(numpy_image)
|
| 314 |
pil_image.save(output_dirs[idx])
|
| 315 |
|
| 316 |
+
# @spaces.GPU
|
| 317 |
def face_enhance(face_image: str, input_image: str, output_image: str, dist_image: str = None, positive_prompt: str = "", id_weight: float = 0.75):
|
| 318 |
initialize_models() # Ensure models are loaded
|
| 319 |
main(face_image, input_image, output_image, dist_image, positive_prompt, id_weight)
|