Update app.py

app.py

@@ -124,7 +124,7 @@ def run_cmd(command):
         print("Process interrupted")
         sys.exit(1)
 
-def run(image,image_u=None):
+def run(image):
     uid = uuid.uuid4()
     
     if os.path.isdir(f"Temp{uid}"):
@@ -133,8 +133,6 @@ def run(image,image_u=None):
     os.makedirs(f"Temp{uid}")
     os.makedirs(f"Temp{uid}/input")
     print(type(image))
-    if image_u != "":
-        image = cv2.imread(image_u)
     cv2.imwrite(f"Temp{uid}/input/input_img.png", image)
 
     command = ("python run.py --input_folder "
@@ -152,10 +150,18 @@ def run(image,image_u=None):
     result_colorization = inferColorization(result_restoration)
 
     return result_colorization
+def load_im(url):
+    return url
+
+    
 with gr.Blocks() as app:
     im = gr.Image(label="Input Image")
-    im_u = gr.Textbox()
+    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)
 app.queue(concurrency_count=100).launch(show_api=False)