Update app.py

app.py

@@ -14,7 +14,7 @@ def load_model():
     device = torch.device("cpu") 
     model = CycleGAN()
     # Загружаем ваш файл (впишите точное имя файла, который вы загрузили!)
-    checkpoint = torch.load("cycle_gan_fixed.pt", map_location=device, weights_only=False)
+    checkpoint = torch.load("cyclegan_150_epochs.pt", map_location=device, weights_only=False)
     model.load_state_dict(checkpoint['model_state_dict'])
     model.eval()
     return model, device