Update app.py

app.py

@@ -126,7 +126,7 @@ def run_cmd(command):
         print("Process interrupted")
         sys.exit(1)
 
-def run(image,txt):
+def run(image):
     
     if os.path.isdir(f"Temp{uid}"):
         shutil.rmtree(f"Temp{uid}")
@@ -139,13 +139,13 @@ def run(image,txt):
     command = ("python run.py --input_folder "
             + f"Temp{uid}/input"
             + " --output_folder "
-            + f"Temp{uid}"
+            + f"Temp"
             + " --GPU "
             + "-1"
             + " --with_scratch")
     run_cmd(command)
 
-    result_restoration = Image.open(f"Temp{uid}/final_output/input_img.png")
+    result_restoration = Image.open(f"Temp/final_output/input_img.png")
     shutil.rmtree(f"Temp{uid}")
     
     result_colorization = inferColorization(result_restoration)
@@ -157,12 +157,7 @@ def load_im(url):
     
 with gr.Blocks() as app:
     im = gr.Image(label="Input Image")
-    with gr.Row():
-        im_u = gr.Textbox()
-        lim_btn=gr.Button("Load")
     im_btn=gr.Button(label="Restore")
     out_im = gr.Image(label="Restored Image")
-
-    #lim_btn(load_im,im_u,im)
-    im_btn.click(run,[im,im_u],out_im)
+    im_btn.click(run,im,out_im)
 app.queue(concurrency_count=100).launch(show_api=False)