Upload TestEnsembleQAPipeline

test_ensemble.py

@@ -22,10 +22,12 @@ class TestEnsembleQAPipeline(Text2TextGenerationPipeline):
 
     def postprocess(self, model_outputs):
         results = {}
+        print(model_outputs)
         for output in model_outputs:
-            guess_text = output['guess']
-            confidence = output['confidence']
-            results = {'guess': guess_text, 'confidence': confidence}
+          print(output)
+          guess_text = output[0]
+          confidence = output[1]
+          results = {'guess': guess_text, 'confidence': confidence}
         return results
 class QuizBowlModel:
     def __init__(self):