VcRlAgent commited on
Commit
03eaae5
·
1 Parent(s): cb8152f

Init Code FaceForge GPU Gradio

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -12,6 +12,12 @@ from PIL import Image, ImageEnhance, ImageOps
12
  from rembg import remove
13
  import io
14
 
 
 
 
 
 
 
15
  # GPU libraries
16
  from gfpgan import GFPGANer
17
  from basicsr.archs.rrdbnet_arch import RRDBNet
 
12
  from rembg import remove
13
  import io
14
 
15
+
16
+ import torchvision
17
+ print("Printing Torch and TorchVision versions:")
18
+ print(torch.__version__)
19
+ print(torchvision.__version__)
20
+
21
  # GPU libraries
22
  from gfpgan import GFPGANer
23
  from basicsr.archs.rrdbnet_arch import RRDBNet