Hugging Face's logo

Spaces:
waveydaveygravy
/
AniPortrait_official
Runtime error

App Files Community
waveydaveygravy commited on
Commit
749c1ac
•
1 Parent(s): 6bc67da

Update face_enhancer.py

Browse files
Files changed (1) hide show
  1. face_enhancer.py +5 -5
face_enhancer.py CHANGED
@@ -22,11 +22,11 @@ def codeformer_runner(img, model):
22
 
23
 
24
  supported_enhancers = {
25
- "CodeFormer": ("./assets/pretrained_models/codeformer.onnx", codeformer_runner),
26
- "GFPGAN": ("./assets/pretrained_models/GFPGANv1.4.pth", gfpgan_runner),
27
- "REAL-ESRGAN 2x": ("./assets/pretrained_models/RealESRGAN_x2.pth", realesrgan_runner),
28
- "REAL-ESRGAN 4x": ("./assets/pretrained_models/RealESRGAN_x4.pth", realesrgan_runner),
29
- "REAL-ESRGAN 8x": ("./assets/pretrained_models/RealESRGAN_x8.pth", realesrgan_runner)
30
  }
31
 
32
  cv2_interpolations = ["LANCZOS4", "CUBIC", "NEAREST"]
 
22
 
23
 
24
  supported_enhancers = {
25
+ "CodeFormer": ("/content/AniPortrait_official/pretrained_model/codeformer.onnx", codeformer_runner),
26
+ "GFPGAN": ("/content/AniPortrait_official/pretrained_model/GFPGANv1.4.pth", gfpgan_runner),
27
+ "REAL-ESRGAN 2x": ("/content/AniPortrait_official/pretrained_model/RealESRGAN_x2.pth", realesrgan_runner),
28
+ "REAL-ESRGAN 4x": ("/content/AniPortrait_official/pretrained_model/RealESRGAN_x4.pth", realesrgan_runner),
29
+ "REAL-ESRGAN 8x": ("/content/AniPortrait_official/pretrained_model/RealESRGAN_x8.pth", realesrgan_runner)
30
  }
31
 
32
  cv2_interpolations = ["LANCZOS4", "CUBIC", "NEAREST"]