abidlabs HF Staff commited on
Commit
0a9c74d
·
1 Parent(s): e5d5416

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -78,10 +78,10 @@ def inference(img, scale, version='v1.3'):
78
  cv2.imwrite(save_path, output)
79
 
80
  output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
81
- return output, save_path
82
  except Exception as error:
83
  print('global exception', error)
84
- return None, None
85
 
86
 
87
  title = "GFPGAN"
@@ -92,7 +92,6 @@ gr.Interface(
92
  gr.inputs.Number(label="Rescaling factor", default=2)
93
  ], [
94
  gr.outputs.Image(type="numpy", label="Output (The whole image)"),
95
- gr.outputs.File(label="Download the output image")
96
  ],
97
  title=title,
98
  examples=[['AI-generate.jpg', 2], ['lincoln.jpg', 2], ['Blake_Lively.jpg', 2],
 
78
  cv2.imwrite(save_path, output)
79
 
80
  output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
81
+ return output
82
  except Exception as error:
83
  print('global exception', error)
84
+ return None
85
 
86
 
87
  title = "GFPGAN"
 
92
  gr.inputs.Number(label="Rescaling factor", default=2)
93
  ], [
94
  gr.outputs.Image(type="numpy", label="Output (The whole image)"),
 
95
  ],
96
  title=title,
97
  examples=[['AI-generate.jpg', 2], ['lincoln.jpg', 2], ['Blake_Lively.jpg', 2],