KenjieDec commited on
Commit
7535eaf
·
verified ·
1 Parent(s): 902ffbe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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(im1, 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,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)")