Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -110,7 +110,7 @@ def inference(file, mode, res_percentage, zoom, x_shift, y_shift):
|
|
110 |
img = cv2.resize(colorf, (grayf.shape[1], grayf.shape[0]))
|
111 |
|
112 |
elif mode == "inpainting":
|
113 |
-
im2 = resize(
|
114 |
model = {'name':'GPEN-Inpainting-1024', 'size':1024}
|
115 |
faceinpainter = FaceInpainting(size=model['size'], model=model['name'], channel_multiplier=2, device='cpu')
|
116 |
im3 = np.asarray(brush_stroke_mask(Image.fromarray(im2)))
|
@@ -163,8 +163,8 @@ with gr.Blocks() as demo:
|
|
163 |
input_img = gr.Image(label="Input Image")
|
164 |
output_img = gr.Image(label="Result")
|
165 |
|
166 |
-
max_res = gr.Slider(1, 200, step=0.5, value=100, label="Output image Resolution Percentage (Higher% = longer processing time)")
|
167 |
restore_type = gr.Radio(["enhance", "colorize", "inpainting", "selfie", "enhanced+background"], value="enhance", type="value", label="Type")
|
|
|
168 |
|
169 |
zoom = gr.Slider(0, 100, step=1, value=50, label="Zoom Percentage (0 = original size)")
|
170 |
x_shift = gr.Slider(0, 100, step=1, label="Horizontal shift Percentage (Before/After)")
|
|
|
110 |
img = cv2.resize(colorf, (grayf.shape[1], grayf.shape[0]))
|
111 |
|
112 |
elif mode == "inpainting":
|
113 |
+
im2 = resize(im, width = 1024)
|
114 |
model = {'name':'GPEN-Inpainting-1024', 'size':1024}
|
115 |
faceinpainter = FaceInpainting(size=model['size'], model=model['name'], channel_multiplier=2, device='cpu')
|
116 |
im3 = np.asarray(brush_stroke_mask(Image.fromarray(im2)))
|
|
|
163 |
input_img = gr.Image(label="Input Image")
|
164 |
output_img = gr.Image(label="Result")
|
165 |
|
|
|
166 |
restore_type = gr.Radio(["enhance", "colorize", "inpainting", "selfie", "enhanced+background"], value="enhance", type="value", label="Type")
|
167 |
+
max_res = gr.Slider(1, 200, step=0.5, value=100, label="Output image Resolution Percentage (Higher% = longer processing time)")
|
168 |
|
169 |
zoom = gr.Slider(0, 100, step=1, value=50, label="Zoom Percentage (0 = original size)")
|
170 |
x_shift = gr.Slider(0, 100, step=1, label="Horizontal shift Percentage (Before/After)")
|