Update app.py

app.py

@@ -9,10 +9,10 @@ import math
 
 
 def greet(input_img, input_model_name, input_tile_mode):
-    if input_img.size[0] * input_img.size[1] > 256 * 256:
-        y = int(math.sqrt(256*256/input_img.size[0]*input_img.size[1]))
-        x = int(input_img.size[0]/input_img.size[1]*y)
-        input_img = ImageOps.fit(input_img, (x, y))
+    # if input_img.size[0] * input_img.size[1] > 256 * 256:
+    #     y = int(math.sqrt(256*256/input_img.size[0]*input_img.size[1]))
+    #     x = int(input_img.size[0]/input_img.size[1]*y)
+    #     input_img = ImageOps.fit(input_img, (x, y))
     input_img = np.array(input_img)
     if input_model_name not in model_cache:
         t1 = time.time()
@@ -42,7 +42,7 @@ if __name__ == '__main__':
     ModelPath = "weights_v3/"
     model_cache = {}
 
-    input_model_name = gr.inputs.Dropdown(os.listdir(ModelPath), default="up2x-latest-no-denoise.pth", label='选择model')
+    input_model_name = gr.inputs.Dropdown(os.listdir(ModelPath), default="up3x-latest-no-denoise", label='选择model')
     input_tile_mode = gr.inputs.Dropdown([0, 1, 2, 3, 4], default=2, label='选择tile_mode')
     input_img = gr.inputs.Image(label='image', type='pil')
 
@@ -53,7 +53,7 @@ if __name__ == '__main__':
                          outputs=outputs,
                          allow_screenshot=False,
                          allow_flagging='never',
-                         examples=[['test-img.jpg', "up2x-latest-no-denoise.pth", 2]],
+                         examples=[['test-img.jpg', "up3x-latest-no-denoise", 2]],
                          article='[https://github.com/bilibili/ailab/tree/main/Real-CUGAN](https://github.com/bilibili/ailab/tree/main/Real-CUGAN)<br>'
                                  '感谢b站开源的项目，图片过大会导致内存不足，所有我将图片裁剪小，想体验大图片的效果请自行前往上面的链接。<br>'
                                  'The large image will lead to memory limit exceeded. So I crop and resize image. '