Spaces:
Build error
Build error
| import cv2 | |
| import os | |
| import cv2 | |
| import shutil | |
| import sys | |
| from subprocess import call | |
| def run_cmd(command): | |
| try: | |
| call(command, shell=True) | |
| except KeyboardInterrupt: | |
| print("Process interrupted") | |
| sys.exit(1) | |
| def Restoration(image): | |
| os.makedirs("Temp") | |
| os.makedirs("Temp/input") | |
| print(type(image)) | |
| cv2.imwrite("Temp/input/input_img.png", image) | |
| command = ("python run.py --input_folder " | |
| + "Temp/input" | |
| + " --output_folder " | |
| + "Temp" | |
| + " --GPU " | |
| + "-1" | |
| + " --with_scratch") | |
| run_cmd(command) | |
| result = cv2.imread("Temp/final_output/input_img.png") | |
| shutil.rmtree("Temp") | |
| return result |