xp3857 commited on
Commit
4e5a93d
·
1 Parent(s): 2084052

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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