fix bug

app.py

@@ -23,11 +23,11 @@ links = """<p style='text-align: center'>
 def call(model_name, sentence, detection_threshold):
 
     pipeline = pipelines[model_name]
-    pipe_out_pred_infos = pipeline([sentence], detection_threshold=detection_threshold)[0]
+    pred_infos = pipeline([sentence], detection_threshold=detection_threshold)[0]
     def pretty_pred_output(pred_info) -> str:
         return "\n".join([f"{qa['question']} --- {';'.join(qa['answers'])}"
                           for qa in pred_info['QAs']])
-    pretty_qa_output = "\n".join(pretty_pred_output(pred_info) for pred_info in pipe_out_pred_infos)
+    pretty_qa_output = "\n".join(pretty_pred_output(pred_info) for pred_info in pred_infos)
     # also present highlighted predicates
     positives = [pred_info['predicate_idx'] for pred_info in pred_infos] 
     def color(idx):
@@ -38,7 +38,7 @@ def call(model_name, sentence, detection_threshold):
         tooltip = f'title=" probability={idx2prob[idx]:.2}&#010;verb={idx2verb[idx]}"' if idx in idx2verb else ''
         return f'<span {tooltip} style="background-color: {color(idx)}">{word}</span>'
     html = '<span>' + ' '.join(word_span(word, idx) for idx, word in enumerate(sentence.split(" "))) + '</span>'
-    return html, pretty_qa_output , pipe_out_pred_infos
+    return html, pretty_qa_output , pred_infos
 
 iface = gr.Interface(fn=call, 
                      inputs=[gr.inputs.Radio(choices=models, default=models[0], label="Model"),