Upload DemoT5QAPipeline

config.json

@@ -6,7 +6,7 @@
   "classifier_dropout": 0.0,
   "custom_pipelines": {
     "demo-t5-small-qa": {
-      "impl": "__main__.DemoT5QAPipeline",
+      "impl": "demo_t5_qa_pipe.DemoT5QAPipeline",
       "pt": [
         "AutoModelForSeq2SeqLM"
       ],

demo_t5_qa_pipe.py

@@ -16,7 +16,7 @@ class DemoT5QAPipeline(Text2TextGenerationPipeline):
             generate_kwargs.get("max_length", self.model.config.max_length),
         )
         outputs = self.model.generate(**model_inputs, **generate_kwargs, return_dict_in_generate=True, output_scores=True, max_new_tokens=75)
-        
+
         # Code from the parent class
         output_ids = outputs.sequences
         out_b = output_ids.shape[0]
@@ -28,7 +28,7 @@ class DemoT5QAPipeline(Text2TextGenerationPipeline):
         output_sequences = outputs.sequences
         output_scores = outputs.scores
         return {"output_ids": output_ids, "output_sequences": output_sequences, "output_scores": output_scores}
-    
+
     def postprocess(self, model_outputs):
         guess_text = super().postprocess(model_outputs)[0]['generated_text']