Update app.py

app.py

@@ -137,10 +137,39 @@ def run(image,Restoration_mode, Colorizaition_mode):
         os.makedirs("Temp/input")
         print(type(image))
         h,w,c=image.shape
+        max = 600
+        if h>w and h > max:
+            r = max / float(h)
+            dim = (int(w * r), max)
+        elif w>h and w > max:
+            r = max / float(w)
+            dim = (max, int(h * r))
+        else:
+            dim = (w,h)
+            pass
+        '''
+        if width is None:
+            # calculate the ratio of the height and construct the
+            # dimensions
+            r = height / float(h)
+            dim = (int(w * r), height)
+    
+        # otherwise, the height is None
+        else:
+            # calculate the ratio of the width and construct the
+            # dimensions
+            r = width / float(w)
+            dim = (width, int(h * r))
+        '''
+
+
+
+
+        
         #if h > 800:
-        h1 = int(h/2)
-        w1 = int(w/2)
-        image=cv2.resize(image, dsize=(h1, w1), interpolation=cv2.INTER_LANCZOS4)
+        #h1 = int(h/2)
+        #w1 = int(w/2)
+        image=cv2.resize(image, dsize=(dim), interpolation=cv2.INTER_LANCZOS4)
         #max_size=(800,800)
         #image=image.thumbnail(max_size)
         #image=np.array(image)