jsaplication commited on
Commit
235ca27
1 Parent(s): 7686b0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,7 +20,7 @@ upsampler = RealESRGANer(scale=4, model_path=model_path, model=model, tile=0, ti
20
  model_gfpgan_1_4 = GFPGANer(model_path='GFPGANv1.4.pth', upscale=2, arch='clean', channel_multiplier=2, bg_upsampler=upsampler)
21
 
22
 
23
- #os.makedirs('output', exist_ok=True)
24
 
25
 
26
  # def inference(img, version, scale, weight):
@@ -63,7 +63,7 @@ def inference(img, version, scale):
63
  extension = 'png'
64
  else:
65
  extension = 'jpg'
66
- save_path = f'out.{extension}'
67
  cv2.imwrite(save_path, output)
68
 
69
  output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
 
20
  model_gfpgan_1_4 = GFPGANer(model_path='GFPGANv1.4.pth', upscale=2, arch='clean', channel_multiplier=2, bg_upsampler=upsampler)
21
 
22
 
23
+ os.makedirs('output', exist_ok=True)
24
 
25
 
26
  # def inference(img, version, scale, weight):
 
63
  extension = 'png'
64
  else:
65
  extension = 'jpg'
66
+ save_path = f'output/out.{extension}'
67
  cv2.imwrite(save_path, output)
68
 
69
  output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)