Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -144,14 +144,14 @@ def run(image):
|
|
144 |
command = ("python run.py --input_folder "
|
145 |
+ f"Temp{uid}/input"
|
146 |
+ " --output_folder "
|
147 |
-
+ f"Temp"
|
148 |
+ " --GPU "
|
149 |
+ "-1"
|
150 |
+ " --with_scratch")
|
151 |
run_cmd(command)
|
152 |
|
153 |
-
result_restoration = Image.open(f"Temp/final_output/input_img.png")
|
154 |
-
shutil.rmtree(f"Temp")
|
155 |
|
156 |
result_colorization = inferColorization(result_restoration)
|
157 |
|
|
|
144 |
command = ("python run.py --input_folder "
|
145 |
+ f"Temp{uid}/input"
|
146 |
+ " --output_folder "
|
147 |
+
+ f"Temp{uid}"
|
148 |
+ " --GPU "
|
149 |
+ "-1"
|
150 |
+ " --with_scratch")
|
151 |
run_cmd(command)
|
152 |
|
153 |
+
result_restoration = Image.open(f"Temp{uid}/final_output/input_img.png")
|
154 |
+
shutil.rmtree(f"Temp{uid}")
|
155 |
|
156 |
result_colorization = inferColorization(result_restoration)
|
157 |
|