Update app.py

app.py

@@ -8,7 +8,7 @@ DEVICE = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
 to_tensor_transform = get_to_tensor_transform(256)
 to_image_transform = get_to_image_transform()
 
-state_dict = torch.load('binaryae_ema_125000.th', map_location='cpu')
+state_dict = torch.load('binaryae_ema_150000.th', map_location='cpu')
 state_dict = {
     k.replace('ae.', '').replace('quantize', 'quantizer').replace('generator', 'decoder'): v
     for k, v in state_dict.items()