xp3857 commited on
Commit
a83cde2
·
1 Parent(s): 04f91a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -151,7 +151,7 @@ def run(image):
151
  + " --with_scratch")
152
  run_cmd(command)
153
 
154
- result_restoration = Image.open("Temp/final_output/input_img.png")
155
  shutil.rmtree("Temp")
156
 
157
  result_colorization = inferColorization(result_restoration)
 
151
  + " --with_scratch")
152
  run_cmd(command)
153
 
154
+ result_restoration = Image.open("./Temp/final_output/input_img.png")
155
  shutil.rmtree("Temp")
156
 
157
  result_colorization = inferColorization(result_restoration)