Update app.py

app.py

@@ -20,7 +20,8 @@ chair="./GEN_chair.pt"
 
 device='cuda' if torch.cuda.is_available() else 'cpu'
 
-ckpt_airplane = torch.load(airplane,map_location=torch.device(device))
+
+ckpt_airplane = torch.load(airplane, map_location=torch.device(device), weights_only=False)
 ckpt_chair = torch.load(chair,map_location=torch.device(device))
 
 def seed_all(seed):