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=32):
-    input_size = int(resolution) * 14
+    input_size = resolution * 14
     data_transforms = transforms.Compose([
         transforms.Resize((input_size, input_size), interpolation=3),
         transforms.ToTensor(),
@@ -68,10 +68,9 @@ attention_interface = gr.Interface(
     fn=get_attention_map,
     inputs=[
         gr.Image(type="pil", label="Input Image"),
-        gr.Dropdown(choices=["16", "32", "64", "128"],
+        gr.Dropdown(choices=[16, 32, 64, 128],
                     label="Attention Map Resolution", 
-                    value="32",
-                    type="index")
+                    value=32)
     ],
     outputs=gr.Image(type="pil", label="SimPool Attention Map", width=width_display, height=height_display),
     examples=example_list,