Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
·
5e8cde2
1
Parent(s):
099b4a7
Update app.py
Browse files
app.py
CHANGED
|
@@ -10,6 +10,10 @@ 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 |
|
| 14 |
def run_cmd(command):
|
| 15 |
try:
|
|
@@ -18,11 +22,7 @@ def run_cmd(command):
|
|
| 18 |
except KeyboardInterrupt:
|
| 19 |
print("Process interrupted")
|
| 20 |
sys.exit(1)
|
| 21 |
-
|
| 22 |
-
run_cmd("pip install basicsr")
|
| 23 |
-
run_cmd("pip install facexlib")
|
| 24 |
-
run_cmd("pip freeze")
|
| 25 |
-
#run_cmd("python setup.py develop")
|
| 26 |
def inference(img):
|
| 27 |
_id = randint(1, 10000)
|
| 28 |
INPUT_DIR = "/tmp/input_image" + str(_id) + "/"
|
|
|
|
| 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 .")
|
| 16 |
+
|
| 17 |
|
| 18 |
def run_cmd(command):
|
| 19 |
try:
|
|
|
|
| 22 |
except KeyboardInterrupt:
|
| 23 |
print("Process interrupted")
|
| 24 |
sys.exit(1)
|
| 25 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
def inference(img):
|
| 27 |
_id = randint(1, 10000)
|
| 28 |
INPUT_DIR = "/tmp/input_image" + str(_id) + "/"
|