wzhouxiff commited on
Commit
bcbb1d2
1 Parent(s): 28474f3

modified app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -66,10 +66,10 @@ def inference(img, version, scale):
66
 
67
  if version == 'RestoreFormer':
68
  face_enhancer = RestoreFormer(
69
- model_path='RestoreFormer.ckpt', upscale=2, arch='RestoreFormer', channel_multiplier=2, bg_upsampler=upsampler)
70
  elif version == 'RestoreFormer++':
71
  face_enhancer = RestoreFormer(
72
- model_path='RestoreFormer++.ckpt', upscale=2, arch='RestoreFormer++', channel_multiplier=2, bg_upsampler=upsampler)
73
 
74
  try:
75
  # _, _, output = face_enhancer.enhance(img, has_aligned=False, only_center_face=False, paste_back=True, weight=weight)
 
66
 
67
  if version == 'RestoreFormer':
68
  face_enhancer = RestoreFormer(
69
+ model_path='RestoreFormer.ckpt', upscale=2, arch='RestoreFormer', bg_upsampler=upsampler)
70
  elif version == 'RestoreFormer++':
71
  face_enhancer = RestoreFormer(
72
+ model_path='RestoreFormer++.ckpt', upscale=2, arch='RestoreFormer++', bg_upsampler=upsampler)
73
 
74
  try:
75
  # _, _, output = face_enhancer.enhance(img, has_aligned=False, only_center_face=False, paste_back=True, weight=weight)