updated handler.py

handler.py

@@ -1,6 +1,7 @@
 from typing import Dict, List, Any
 import torch
 import torch.nn as nn
+import json
 from transformers import pipeline, BertModel, AutoTokenizer, PretrainedConfig
 
 class EndpointHandler():
@@ -28,6 +29,7 @@ class EndpointHandler():
 
         # run normal prediction
         prediction = self.model.classify(inputs)
+        prediction = json.dumps(prediction)
         return prediction
     
 class CustomModel(nn.Module):