ohayonguy commited on
Commit
beeddbb
1 Parent(s): 9a7cbfe

trying to fix run

Browse files
Files changed (1) hide show
  1. app.py +16 -18
app.py CHANGED
@@ -115,24 +115,22 @@ def inference(img, aligned, scale, num_steps):
115
  device=device,
116
  model_rootpath=None)
117
 
118
- try:
119
- has_aligned = True if aligned == 'aligned' else False
120
- _, restored_aligned, restored_img = enhance_face(img, face_helper, has_aligned, only_center_face=False,
121
- paste_back=True)
122
- if has_aligned:
123
- output = restored_aligned[0]
124
- else:
125
- output = restored_img
126
- except RuntimeError as error:
127
- print('Error', error)
128
 
129
- try:
130
- if scale != 2:
131
- interpolation = cv2.INTER_AREA if scale < 2 else cv2.INTER_LANCZOS4
132
- h, w = img.shape[0:2]
133
- output = cv2.resize(output, (int(w * scale / 2), int(h * scale / 2)), interpolation=interpolation)
134
- except Exception as error:
135
- print('Wrong scale input.', error)
 
136
  if img_mode == 'RGBA': # RGBA images should be saved in png format
137
  extension = 'png'
138
  else:
@@ -154,7 +152,7 @@ demo = gr.Interface(
154
  inference, [
155
  gr.Image(type="filepath", label="Input"),
156
  gr.Radio(['aligned', 'unaligned'], type="value", value='unaligned', label='Image Alignment'),
157
- gr.Number(label="Rescaling factor", value=2),
158
  gr.Number(label="Number of flow steps (a higher value leads to better image quality at the expense of runtime)", value=25),
159
  ], [
160
  gr.Image(type="numpy", label="Output (The whole image)"),
 
115
  device=device,
116
  model_rootpath=None)
117
 
118
+ has_aligned = True if aligned == 'aligned' else False
119
+ _, restored_aligned, restored_img = enhance_face(img, face_helper, has_aligned, only_center_face=False,
120
+ paste_back=True)
121
+ if has_aligned:
122
+ output = restored_aligned[0]
123
+ else:
124
+ output = restored_img
 
 
 
125
 
126
+
127
+ # try:
128
+ # if scale != 2:
129
+ # interpolation = cv2.INTER_AREA if scale < 2 else cv2.INTER_LANCZOS4
130
+ # h, w = img.shape[0:2]
131
+ # output = cv2.resize(output, (int(w * scale / 2), int(h * scale / 2)), interpolation=interpolation)
132
+ # except Exception as error:
133
+ # print('Wrong scale input.', error)
134
  if img_mode == 'RGBA': # RGBA images should be saved in png format
135
  extension = 'png'
136
  else:
 
152
  inference, [
153
  gr.Image(type="filepath", label="Input"),
154
  gr.Radio(['aligned', 'unaligned'], type="value", value='unaligned', label='Image Alignment'),
155
+ gr.Number(label="Rescaling factor (the rescaling factor of the final image)", value=2),
156
  gr.Number(label="Number of flow steps (a higher value leads to better image quality at the expense of runtime)", value=25),
157
  ], [
158
  gr.Image(type="numpy", label="Output (The whole image)"),