Update app.py

app.py

@@ -136,8 +136,13 @@ def run(image,Restoration_mode, Colorizaition_mode):
         os.makedirs("Temp")
         os.makedirs("Temp/input")
         print(type(image))
-        max_size=(800,800)
-        image=image.thumbnail(max_size)
+        h,w,c=image.shape
+        #if h > 800:
+        h1 = int(h/2)
+        w1 = int(w/2)
+        image=cv2.resize(image, dsize=(h1, w1), interpolation=cv2.INTER_LANCZOS4)
+        #max_size=(800,800)
+        #image=image.thumbnail(max_size)
         #image=np.array(image)
         image=cv2.imread(image)
         cv2.imwrite("Temp/input/input_img.png", image)
@@ -167,7 +172,7 @@ with gr.Blocks() as app:
     with gr.Row():
         gr.Column()
         with gr.Column():
-            im = gr.Pil(label="Input Image")
+            im = gr.Image(label="Input Image")
             rad1 = gr.Radio(["BOPBTL", "Pix2Pix"])
             rad2 = gr.Radio(["Deoldify", "Pix2Pix Resnet 9block","Pix2Pix Unet 256"])