Update handler.py

handler.py

@@ -59,6 +59,10 @@ class EndpointHandler:
             "hate_speech": decoded_hate_speech
         }
 
+    def load_model(self, model_path):
+        #Load model weights from the specified path
+        self.load_state_dict(torch.load(model_path))
+
 # Define your custom multitask model architecture here
 class MultitaskDebertaModel(nn.Module):
     def __init__(self, num_emotion_labels, num_polarity_labels, num_hate_speech_labels):
@@ -142,7 +146,3 @@ class MultitaskDebertaModel(nn.Module):
             'polarity': polarity_logits,
             'hate_speech': hate_speech_logits
         }
-
-        def load_model(self, model_path):
-        #Load model weights from the specified path
-        self.load_state_dict(torch.load(model_path))