higher resolution choice

app.py

@@ -46,7 +46,7 @@ msg = model.load_state_dict(state_dict, strict=True)
 model.eval()
 
 def get_attention_map(img, resolution):
-    input_size = resolution * 14
+    input_size = int(resolution) * 14
     data_transforms = transforms.Compose([
         transforms.Resize((input_size, input_size), interpolation=3),
         transforms.ToTensor(),
@@ -66,9 +66,13 @@ def get_attention_map(img, resolution):
 
 attention_interface = gr.Interface(
     fn=get_attention_map,
-    inputs=[gr.Image(type="pil", label="Input Image"), 
-            gr.Dropdown(choices=["16", "32", "64", "128"],
-                        label="Attention Map Resolution", value="32", type="index")],
+    inputs=[
+        gr.Image(type="pil", label="Input Image"),
+        gr.Dropdown(choices=["16", "32", "64", "128"],
+                    label="Attention Map Resolution", 
+                    value="32",
+                    type="index")
+    ],
     outputs=gr.Image(type="pil", label="SimPool Attention Map", width=width_display, height=height_display),
     examples=example_list,
     title="Explore the Attention Maps of SimPool🔍",