Update app.py

app.py

@@ -18,7 +18,7 @@ def predict_label(text):
     ensemble_span_scores = [score for scores in [span_scores, msa_span_scores] for score in scores]
     ensemble_pooled_scores = pool_span_scores(ensemble_span_scores, ip_len)
     ensemble_pred_tags = [entities_from_token_classes(sent_targs) for sent_targs in ensemble_pooled_scores]
-
+    print('ensemble_pred_tags: ', ensemble_pred_tags)
     ent_scores = extract_ent_scores(entity_model,ip,ensemble_pred_tags, pos_col=1, task_col=2)
     combined_sequences, ent_pred_tags = pool_ent_scores(ent_scores, ip_len)
     
@@ -48,6 +48,6 @@ if __name__ == '__main__':
     span_model = SpanNet.load_model(span_path)
     msa_span_model = SpanNet.load_model(msa_span_path)
     entity_model = EntNet.load_model(entity_path)
-    
+    iface= gr.base(primary_hue="green")
     iface = gr.Interface(fn=predict_label, inputs="text", outputs="text")
     iface.launch(show_api=False)