Update pipeline.py

pipeline.py

@@ -35,7 +35,7 @@ class PreTrainedPipeline():
     def __init__(self, path=""):
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         self.model = Generator().to(self.device)
-        self.model.load_state_dict(torch.load("pytorch_model.bin", map_location=device))
+        self.model.load_state_dict(torch.load("pytorch_model.bin", map_location=self.device))
  
     def __call__(self, inputs: str):
         """