Update inference_gfpgan.py
Browse files- 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} ...')
|