Update handler.py

handler.py

@@ -34,7 +34,7 @@ class EndpointHandler:
     def __init__(self, path=""):
         # Cargar el modelo
         self.model = MiniGPT()
-        checkpoint = os.path.join(path, "mini_gpt.pth")
+        checkpoint = os.path.join(path, "pytorch_model.bin")
         self.model.load_state_dict(torch.load(checkpoint, map_location="cpu"))
         self.model.eval()