Update app.py

app.py

@@ -18,7 +18,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 state_dict: dict = torch.hub.load_state_dict_from_url(
     "https://www.robots.ox.ac.uk/~vgg/research/selfmask/shared_files/selfmask_nq20.pt",
     map_location=device  # "cuda" if torch.cuda.is_available() else "cpu"
-)["model"]
+)
 
 parser = ArgumentParser("SelfMask demo")
 parser.add_argument(