Update app.py

app.py

@@ -172,11 +172,11 @@ tag_model = ram(pretrained='preset/models/ram_swin_large_14m.pth',
 tag_model.eval()
 tag_model.to(device, dtype=weight_dtype)
 
-def preprocess_image(input_image: Image.Image):
-    input_image = input_image.resize((256, 256), Image.Resampling.BILINEAR)
-
-    return input_image
-
+def preprocess_image(input_image: Image.Image) -> Image.Image:
+    img = input_image.copy()
+    img.thumbnail((256, 256), Image.Resampling.BILINEAR)
+    return img
+    
 @spaces.GPU()
 def process(
     input_image: Image.Image,
@@ -312,6 +312,10 @@ with gr.Blocks(css=css) as demo:
                         [
                             "preset/datasets/test_datasets/apologise.png", 
                         ], 
+                        [
+                            "preset/datasets/test_datasets/gummibears.png", 
+                        ], 
+                        
                     ],
                     inputs=[
                         input_image,