fix the resize pre-process

app.py

@@ -70,10 +70,10 @@ def process_image(prompt_img, query_img, options):
     query_img = pil_to_tensor(query_img).float() / 255.0
     prompt_img = pil_to_tensor(prompt_img).float() / 255.0
     
-    if query_img.shape[1] != img_size:
-        resize_trans = K.augmentation.Resize([img_size, img_size], return_transform=True)
-        query_img = resize_trans(query_img)[0]
-        prompt_img = resize_trans(prompt_img)[0]
+    # Resize image
+    resize_trans = K.augmentation.Resize([img_size, img_size], return_transform=True)
+    query_img = resize_trans(query_img)[0]
+    prompt_img = resize_trans(prompt_img)[0]
 
     
     test_data = {
@@ -110,6 +110,8 @@ examples = [
     ["images/134.png", "images/000.png", "model-256"],
     ["images/036.png", "images/024.png", "model-256"],
     ["images/178.png", "images/003.png", "model-256"],
+    ["images/img1-02.jpg",  "images/img1-03.jpg", "model-256"]
+    ["images/img2-03.jpg",  "images/img2-03.jpg", "model-256"]
 ]
 
 # Gradio interface layout