Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -65,18 +65,18 @@ def inference(img, version, scale):
|
|
65 |
if h < 300:
|
66 |
img = cv2.resize(img, (w * 2, h * 2), interpolation=cv2.INTER_LANCZOS4)
|
67 |
|
68 |
-
|
69 |
-
|
70 |
-
|
71 |
elif version == 'v1.3':
|
72 |
face_enhancer = GFPGANer(
|
73 |
model_path='GFPGANv1.3.pth', upscale=2, arch='clean', channel_multiplier=2, bg_upsampler=upsampler)
|
74 |
-
|
75 |
-
|
76 |
-
|
77 |
-
|
78 |
-
|
79 |
-
|
80 |
# elif version == 'CodeFormer':
|
81 |
# face_enhancer = GFPGANer(
|
82 |
# model_path='CodeFormer.pth', upscale=2, arch='CodeFormer', channel_multiplier=2, bg_upsampler=upsampler)
|
@@ -126,7 +126,7 @@ demo = gr.Interface(
|
|
126 |
inference, [
|
127 |
gr.inputs.Image(type="filepath", label="Input"),
|
128 |
# gr.inputs.Radio(['v1.2', 'v1.3', 'v1.4', 'RestoreFormer', 'CodeFormer'], type="value", default='v1.4', label='version'),
|
129 |
-
gr.inputs.Radio(['v1.
|
130 |
gr.inputs.Number(label="Rescaling factor", default=2),
|
131 |
# gr.Slider(0, 100, label='Weight, only for CodeFormer. 0 for better quality, 100 for better identity', default=50)
|
132 |
], [
|
|
|
65 |
if h < 300:
|
66 |
img = cv2.resize(img, (w * 2, h * 2), interpolation=cv2.INTER_LANCZOS4)
|
67 |
|
68 |
+
# if version == 'v1.2':
|
69 |
+
## face_enhancer = GFPGANer(
|
70 |
+
# model_path='GFPGANv1.2.pth', upscale=2, arch='clean', channel_multiplier=2, bg_upsampler=upsampler)
|
71 |
elif version == 'v1.3':
|
72 |
face_enhancer = GFPGANer(
|
73 |
model_path='GFPGANv1.3.pth', upscale=2, arch='clean', channel_multiplier=2, bg_upsampler=upsampler)
|
74 |
+
# elif version == 'v1.4':
|
75 |
+
# face_enhancer = GFPGANer(
|
76 |
+
# model_path='GFPGANv1.4.pth', upscale=2, arch='clean', channel_multiplier=2, bg_upsampler=upsampler)
|
77 |
+
# elif version == 'RestoreFormer':
|
78 |
+
# face_enhancer = GFPGANer(
|
79 |
+
# model_path='RestoreFormer.pth', upscale=2, arch='RestoreFormer', channel_multiplier=2, bg_upsampler=upsampler)
|
80 |
# elif version == 'CodeFormer':
|
81 |
# face_enhancer = GFPGANer(
|
82 |
# model_path='CodeFormer.pth', upscale=2, arch='CodeFormer', channel_multiplier=2, bg_upsampler=upsampler)
|
|
|
126 |
inference, [
|
127 |
gr.inputs.Image(type="filepath", label="Input"),
|
128 |
# gr.inputs.Radio(['v1.2', 'v1.3', 'v1.4', 'RestoreFormer', 'CodeFormer'], type="value", default='v1.4', label='version'),
|
129 |
+
gr.inputs.Radio(['v1.4'], type="value", default='v1.4', label='version'),
|
130 |
gr.inputs.Number(label="Rescaling factor", default=2),
|
131 |
# gr.Slider(0, 100, label='Weight, only for CodeFormer. 0 for better quality, 100 for better identity', default=50)
|
132 |
], [
|