victorisgeek commited on
Commit
9cfb37a
1 Parent(s): d27d214

Update face_enhancer.py

Browse files
Files changed (1) hide show
  1. face_enhancer.py +3 -5
face_enhancer.py CHANGED
@@ -10,12 +10,11 @@ def gfpgan_runner(img, model):
10
  _, imgs, _ = model.enhance(img, paste_back=True, has_aligned=True)
11
  return imgs[0]
12
 
13
-
14
  def realesrgan_runner(img, model):
15
  img = model.predict(img)
16
  return img
17
 
18
- def realhatgan_runner(img, model):
19
  img = model.predict(img)
20
  return img
21
 
@@ -23,11 +22,10 @@ def restoreformer_runner(img, model):
23
  img = model.predict(img)
24
  return img
25
 
26
-
27
- def ultrasharp_runner(img, model):
28
  img = model.predict(img)
29
  return img
30
-
31
  def codeformer_runner(img, model):
32
  img = model.enhance(img)
33
  return img
 
10
  _, imgs, _ = model.enhance(img, paste_back=True, has_aligned=True)
11
  return imgs[0]
12
 
 
13
  def realesrgan_runner(img, model):
14
  img = model.predict(img)
15
  return img
16
 
17
+ def realhatgan_runner(img, model):
18
  img = model.predict(img)
19
  return img
20
 
 
22
  img = model.predict(img)
23
  return img
24
 
25
+ def ultrasharp_runner(img, model):
 
26
  img = model.predict(img)
27
  return img
28
+
29
  def codeformer_runner(img, model):
30
  img = model.enhance(img)
31
  return img