Upload pipeline.py

pipeline.py

@@ -81,8 +81,8 @@ class PreTrainedPipeline():
         # process input
         inputs = data.pop("inputs", data)
         # process input text
-        prediction = self.predict(inputs, self.model, self.tokenizer)
-        return {"text":prediction}
+        prediction = predict(inputs, self.model, self.tokenizer)
+        return {"generated_text": prediction}