Update app.py

app.py

@@ -124,7 +124,7 @@ def run_cmd(command):
         print("Process interrupted")
         sys.exit(1)
 
-def run(image,Restoration_mode, Colorizaition_mode):
+def run(image,image_u=None):
     uid = uuid.uuid4()
     Restoration_mode == "BOPBTL"
     if os.path.isdir(f"Temp{uid}"):
@@ -133,6 +133,8 @@ def run(image,Restoration_mode, Colorizaition_mode):
     os.makedirs(f"Temp{uid}")
     os.makedirs(f"Temp{uid}/input")
     print(type(image))
+    if image_u != "":
+        image = cv2.imread(f"{image_u}")
     cv2.imwrite(f"Temp{uid}/input/input_img.png", image)
 
     command = ("python run.py --input_folder "
@@ -151,9 +153,9 @@ def run(image,Restoration_mode, Colorizaition_mode):
 
     return result_colorization
 with gr.Blocks() as app:
-    #im = gr.Image(label="Input Image")
-    im = gr.Textbox()
+    im = gr.Image(label="Input Image")
+    im_u = gr.Textbox()
     im_btn=gr.Button(label="Restore")
     out_im = gr.Image(label="Restored Image")
-    im_btn.click(run,im,out_im)
+    im_btn.click(run,[im,im_u],out_im)
 app.queue(concurrency_count=100).launch(show_api=False)