Hugging Face's logo

Spaces:
victorisgeek
/
EG
Running

App Files Community
victorisgeek commited on
Commit
bcf6b14
•
1 Parent(s): 9cfb37a

Update face_enhancer.py

Browse files
Files changed (1) hide show
  1. face_enhancer.py +1 -1
face_enhancer.py CHANGED
@@ -62,7 +62,7 @@ def load_face_enhancer_model(name='GFPGAN', device="cpu"):
62
  model = CodeFormerEnhancer(model_path=model_path, device=device)
63
  elif name == 'GFPGAN':
64
  model = gfpgan.GFPGANer(model_path=model_path, upscale=1, device=device)
65
- elif name ='RestoreFormer':
66
  model = RestoreFormer(device, scale=1)
67
  model.load_weights(model_path, download=False)
68
  elif name == 'REAL-ESRGAN 4x':
 
62
  model = CodeFormerEnhancer(model_path=model_path, device=device)
63
  elif name == 'GFPGAN':
64
  model = gfpgan.GFPGANer(model_path=model_path, upscale=1, device=device)
65
+ elif name -='RestoreFormer':
66
  model = RestoreFormer(device, scale=1)
67
  model.load_weights(model_path, download=False)
68
  elif name == 'REAL-ESRGAN 4x':