Spaces:
Running
Running
set proper space name
Browse files- .gitignore +10 -1
- app.py +10 -2
.gitignore
CHANGED
|
@@ -1 +1,10 @@
|
|
| 1 |
-
.idea
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
.idea
|
| 2 |
+
facelib/detection/__pycache__/
|
| 3 |
+
facelib/detection/retinaface/__pycache__/
|
| 4 |
+
facelib/detection/yolov5face/__pycache__/
|
| 5 |
+
facelib/detection/yolov5face/models/__pycache__/
|
| 6 |
+
facelib/detection/yolov5face/utils/__pycache__/
|
| 7 |
+
facelib/parsing/__pycache__/
|
| 8 |
+
facelib/utils/__pycache__/
|
| 9 |
+
models/__pycache__/
|
| 10 |
+
utils/__pycache__/
|
app.py
CHANGED
|
@@ -68,6 +68,14 @@ face_helper = FaceRestoreHelper(
|
|
| 68 |
device=device)
|
| 69 |
|
| 70 |
def predict(image):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 71 |
face_helper.clean_all()
|
| 72 |
face_helper.read_image(image)
|
| 73 |
face_helper.get_face_landmarks_5(only_center_face=False, resize=640, eye_dist_threshold=5)
|
|
@@ -100,9 +108,9 @@ def predict(image):
|
|
| 100 |
return image, restored_img
|
| 101 |
|
| 102 |
|
| 103 |
-
with gr.Blocks(title="
|
| 104 |
navbar = gr.Navbar(visible=True, main_page_name="Workspace")
|
| 105 |
-
gr.Markdown("##
|
| 106 |
with gr.Row():
|
| 107 |
with gr.Column(scale=1):
|
| 108 |
with gr.Row():
|
|
|
|
| 68 |
device=device)
|
| 69 |
|
| 70 |
def predict(image):
|
| 71 |
+
"""
|
| 72 |
+
Enhances the image face.
|
| 73 |
+
|
| 74 |
+
Parameters:
|
| 75 |
+
image (string): File path to the input image.
|
| 76 |
+
Returns:
|
| 77 |
+
image (string): paths for image face enhanced.
|
| 78 |
+
"""
|
| 79 |
face_helper.clean_all()
|
| 80 |
face_helper.read_image(image)
|
| 81 |
face_helper.get_face_landmarks_5(only_center_face=False, resize=640, eye_dist_threshold=5)
|
|
|
|
| 108 |
return image, restored_img
|
| 109 |
|
| 110 |
|
| 111 |
+
with gr.Blocks(title="CodeFormer") as app:
|
| 112 |
navbar = gr.Navbar(visible=True, main_page_name="Workspace")
|
| 113 |
+
gr.Markdown("## CodeFormer")
|
| 114 |
with gr.Row():
|
| 115 |
with gr.Column(scale=1):
|
| 116 |
with gr.Row():
|