Update app.py

app.py

@@ -146,7 +146,7 @@ def run(image):
     run_cmd(command)
 
     result_restoration = Image.open(f"Temp/final_output/input_img.png")
-    shutil.rmtree(f"Temp{uid}")
+    shutil.rmtree(f"Temp")
     
     result_colorization = inferColorization(result_restoration)
 
@@ -156,8 +156,12 @@ def load_im(url):
  
     
 with gr.Blocks() as app:
-    im = gr.Image(label="Input Image")
-    im_btn=gr.Button(label="Restore")
-    out_im = gr.Image(label="Restored Image")
+    with gr.Row():
+        gr.Column()
+        with gr.Column():
+            im = gr.Image(label="Input Image")
+            im_btn=gr.Button(label="Restore")
+            out_im = gr.Image(label="Restored Image")
+        gr.Column()
     im_btn.click(run,im,out_im)
 app.queue(concurrency_count=100).launch(show_api=False)