Center crop images and resize

app.py

@@ -8,6 +8,7 @@ import gradio as gr
 import pandas as pd
 from PIL import Image
 from braceexpand import braceexpand
+from torchvision import transforms
 
 
 # Load model
@@ -165,6 +166,12 @@ def retrieve(query):
         with tarfile.open("files/combined.tar", "r") as tar_file:
             image = tar_file.extractfile(f"{image_name}.jpg")
             image = Image.open(image).copy()
+            # Center crop the image
+            width, height = image.size
+            new_size = min(width, height)
+            image = transforms.CenterCrop(new_size)(image)
+            # Resize the image
+            image = transforms.Resize((600, 600))(image)
             images.append(image)
     return images