Update app.py

app.py

@@ -37,7 +37,7 @@ def get_attention_models() -> List[str]:
     """Get a list of timm models that have attention blocks."""
     all_models = timm.list_pretrained()
     # FIXME Focusing on ViT models for initial impl
-    attention_models = [model for model in all_models if any([model.lower().startswith(p) for p in ('vit', 'deit', 'beit', 'eva'))]
+    attention_models = [model for model in all_models if any([model.lower().startswith(p) for p in ('vit', 'deit', 'beit', 'eva')])]
     return attention_models
 
 def load_model(model_name: str) -> Tuple[torch.nn.Module, AttentionExtractor]: