Update app.py

app.py

@@ -23,17 +23,16 @@ links = """<p style='text-align: center'>
 def call(sentence, detection_threshold):
 
     outputs = pipeline([sentence], nominalization_detection_threshold=detection_threshold)[0]
-    def pretty_qadisc_qas(pred_info) -> List[str]:
-        if not pred_info: return []
+    def pretty_qadisc_qas(qa_infos) -> List[str]:
+        if not qa_infos: return []
         return [f"{qa['question']} --- {qa['answer']}"
-                for qa in pred_info if qa is not None]
+                for qa in qa_infos if qa is not None]
     def pretty_qasrl_qas(pred_info) -> List[str]:
         if not pred_info or not pred_info['QAs']: return []
         return [f"{qa['question']} --- {';'.join(qa['answers'])}"
                 for qa in pred_info['QAs'] if qa is not None]
-    
-    qasrl_qas = pretty_qasrl_qas(outputs['qasrl'])
-    qanom_qas = pretty_qasrl_qas(outputs['qanom'])
+    qasrl_qas = [qa for pred_info in outputs['qasrl'] for qa in pretty_qasrl_qas(pred_info)]
+    qanom_qas = [qa for pred_info in outputs['qanom'] for qa in pretty_qasrl_qas(pred_info)]
     qadisc_qas= pretty_qadisc_qas(outputs['qadiscourse'])
     all_qas = ['QASRL:'] + qasrl_qas + ['\nQANom:'] + qanom_qas + ['\nQADiscourse:'] + qadisc_qas                      
     if not qasrl_qas + qanom_qas + qadisc_qas: