Update handler.py

handler.py

@@ -5,7 +5,7 @@ from transformers import pipeline
 class EndpointHandler:
     def __init__(self, path=""):
         self.pipeline = pipeline(
-            "text-generation",
+            task="text-generation",
             model="mistralai/Mixtral-8x7B-Instruct-v0.1",
             device_map='auto',
             #trust_remote_code=True,
@@ -24,9 +24,7 @@ class EndpointHandler:
         Return:
               A :obj:`list` | `dict`: will be serialized and returned
         """
-        # get inputs
         inputs = data.pop("inputs", "")
-        # get additional date field
         params = data.pop("parameters", ())
         if not params:
             params = dict()