Lin0He
/

text-summary-gpt2-short

text-generation

text-generation-inference

Model card Files Files and versions

Lin0He commited on Dec 5, 2023

Commit

75a3e98

·

1 Parent(s): 8a6a2c0

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

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

         # process input
         inputs = data.pop("inputs", data)
         # process input text
+        prediction = predict(inputs['text'], self.model, self.tokenizer)
         return prediction #{"generated_text": prediction}