leonelhs commited on
Commit
1db0aa8
·
1 Parent(s): 699c2ed

set proper space name

Browse files
Files changed (2) hide show
  1. .gitignore +10 -1
  2. 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="RestoreFormer") as app:
104
  navbar = gr.Navbar(visible=True, main_page_name="Workspace")
105
- gr.Markdown("## RestoreFormer")
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():