Update app.py

app.py

@@ -42,7 +42,11 @@ def call(sentence, layers, detection_threshold):
     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                      
+    all_qas = []
+    if "qasrl" in layers: all_qas += ['\nQASRL:'] + qasrl_qas
+    if "qanom" in layers: all_qas += ['\nQANom:'] + qanom_qas
+    if "qadiscourse" in layers: all_qas += ['\nQADiscourse:'] + qadisc_qas
+
     if not qasrl_qas + qanom_qas + qadisc_qas:
         pretty_qa_output = "NO QA GENERATED"
     else: