Update handler.py

handler.py

@@ -1,6 +1,7 @@
 import torch
 import torch.nn as nn
 from huggingface_hub import PyTorchModelHubMixin
+import pandas as pd
 
 # Set device
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')  
@@ -25,7 +26,13 @@ class EndpointHandler:
     def __call__(self, data):
 
         print(f"Payload: {data}")
-        new_data_tensor = torch.tensor(data['chargeData'], dtype=torch.float32).to(device)  # Ensure tensor is on device
+
+        # Create a Pandas DataFrame
+        payloadDataFrame = pd.DataFrame(data['chargeData'])
+        print(payloadDataFrame)
+
+        
+        new_data_tensor = torch.tensor(payloadDataFrame.values, dtype=torch.float).to(device)  # Ensure tensor is on device
 
         # Make predictions
         with torch.no_grad():