account for aspect ratio

app.py

@@ -42,7 +42,11 @@ def generate_relevancy(
     if len(labels) > 32:
         labels = labels[:32]
     prompts = [prompt]
-    img = np.asarray(Image.fromarray(img).resize((244 * 4, 244 * 4)))
+    resize_shape = np.array(img.shape[:2])
+    resize_shape = tuple(
+        ((resize_shape / resize_shape.max()) * 224 * 4).astype(int).tolist()
+    )
+    img = np.asarray(Image.fromarray(img).resize(resize_shape))
     assert img.dtype == np.uint8
     h, w, c = img.shape
     start = time()
@@ -95,7 +99,7 @@ This relevancy extractor builds heavily on [Chefer et al.'s codebase](https://gi
         gr.Image(type="numpy", label="Image"),
         gr.Textbox(label="Labels (comma separated without spaces in between)"),
         gr.Textbox(
-            label="Prompt. (Make sure to include \{\} in the prompt like examples below)"
+            label="Prompt. (Make sure to include '{}' in the prompt like examples below)"
         ),
         gr.Dropdown(
             value="ours",