renatotn7 commited on
Commit
1a07e5b
·
1 Parent(s): 79ab753

Update inference_gfpgan.py

Browse files
Files changed (1) hide show
  1. inference_gfpgan.py +2 -0
inference_gfpgan.py CHANGED
@@ -4,6 +4,7 @@ import glob
4
  import numpy as np
5
  import os
6
  import torch
 
7
  from basicsr.utils import imwrite
8
 
9
  from gfpgan import GFPGANer
@@ -121,6 +122,7 @@ def main():
121
 
122
  # ------------------------ restore ------------------------
123
  for img_path in img_list:
 
124
  # read image
125
  img_name = os.path.basename(img_path)
126
  print(f'Processing {img_name} ...')
 
4
  import numpy as np
5
  import os
6
  import torch
7
+ import streamlit as st
8
  from basicsr.utils import imwrite
9
 
10
  from gfpgan import GFPGANer
 
122
 
123
  # ------------------------ restore ------------------------
124
  for img_path in img_list:
125
+ st.write(img_path)
126
  # read image
127
  img_name = os.path.basename(img_path)
128
  print(f'Processing {img_name} ...')