Ahsen Khaliq commited on
Commit
f25bddf
1 Parent(s): f6e58fd
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -10,6 +10,15 @@ import psutil
10
  torch.hub.download_url_to_file('https://upload.wikimedia.org/wikipedia/commons/thumb/a/ab/Abraham_Lincoln_O-77_matte_collodion_print.jpg/1024px-Abraham_Lincoln_O-77_matte_collodion_print.jpg', 'lincoln.jpg')
11
  torch.hub.download_url_to_file('https://upload.wikimedia.org/wikipedia/commons/5/50/Albert_Einstein_%28Nobel%29.png', 'einstein.png')
12
 
 
 
 
 
 
 
 
 
 
13
  os.system("pip install gfpgan")
14
  os.system("pip freeze")
15
  os.system("wget https://github.com/TencentARC/GFPGAN/releases/download/v0.2.0/GFPGANCleanv1-NoCE-C2.pth -P .")
 
10
  torch.hub.download_url_to_file('https://upload.wikimedia.org/wikipedia/commons/thumb/a/ab/Abraham_Lincoln_O-77_matte_collodion_print.jpg/1024px-Abraham_Lincoln_O-77_matte_collodion_print.jpg', 'lincoln.jpg')
11
  torch.hub.download_url_to_file('https://upload.wikimedia.org/wikipedia/commons/5/50/Albert_Einstein_%28Nobel%29.png', 'einstein.png')
12
 
13
+ import argparse
14
+ import cv2
15
+ import glob
16
+ import numpy as np
17
+ import os
18
+ import torch
19
+ from basicsr.utils import imwrite
20
+ from gfpgan import GFPGANer
21
+
22
  os.system("pip install gfpgan")
23
  os.system("pip freeze")
24
  os.system("wget https://github.com/TencentARC/GFPGAN/releases/download/v0.2.0/GFPGANCleanv1-NoCE-C2.pth -P .")