Update pipeline.py

pipeline.py

@@ -34,7 +34,7 @@ class PretrainedPipeline():
     def __init__(self):
         self.device = torch.device("cpu")
         self.generator = Generator()  # Instantiate your GAN generator class
-        self.generator.load_state_dict(torch.load("pytorch_model.bin", map_location=self.device))
+        self.generator.load_state_dict(torch.load("generator.pth", map_location=self.device))
         self.generator.eval()
 
     def generate_image(self):