abidlabs HF Staff commited on
Commit
3586b92
·
1 Parent(s): 8d2c992

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -39,7 +39,7 @@ face_enhancer_v2 = GFPGANer(
39
  os.makedirs('output', exist_ok=True)
40
 
41
 
42
- def inference(img, version, scale):
43
  print(img, version, scale)
44
  try:
45
  img = cv2.imread(img, cv2.IMREAD_UNCHANGED)
@@ -90,7 +90,6 @@ description = ""
90
  gr.Interface(
91
  inference, [
92
  gr.inputs.Image(type="filepath", label="Input"),
93
- gr.inputs.Radio(['v1.2', 'v1.3'], type="value", default='v1.3', label='GFPGAN version'),
94
  gr.inputs.Number(label="Rescaling factor", default=2)
95
  ], [
96
  gr.outputs.Image(type="numpy", label="Output (The whole image)"),
 
39
  os.makedirs('output', exist_ok=True)
40
 
41
 
42
+ def inference(img, scale, version='v1.3'):
43
  print(img, version, scale)
44
  try:
45
  img = cv2.imread(img, cv2.IMREAD_UNCHANGED)
 
90
  gr.Interface(
91
  inference, [
92
  gr.inputs.Image(type="filepath", label="Input"),
 
93
  gr.inputs.Number(label="Rescaling factor", default=2)
94
  ], [
95
  gr.outputs.Image(type="numpy", label="Output (The whole image)"),