Lin0He commited on
Commit
f928cbd
·
1 Parent(s): be51c66

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -77,7 +77,7 @@ class PreTrainedPipeline():
77
  self.model = AutoModelForCausalLM.from_pretrained("Lin0He/text-summary-gpt2-short")
78
 
79
 
80
- def __call__(self, Dict[Any, Any]) -> Dict[Any, Any]:
81
  # process input
82
  inputs = data.pop("inputs", data)
83
  # process input text
 
77
  self.model = AutoModelForCausalLM.from_pretrained("Lin0He/text-summary-gpt2-short")
78
 
79
 
80
+ def __call__(self, data:Dict[Any, Any]) -> Dict[Any, Any]:
81
  # process input
82
  inputs = data.pop("inputs", data)
83
  # process input text