kowalsky commited on
Commit
e409316
·
1 Parent(s): a950efa

fixed the predict function

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -54,8 +54,7 @@ def predict(input_data):
54
  labels = intent_labels_to_ids()
55
  intent_labels = intent_ids_to_labels(labels)
56
 
57
- print(f"Ner logits: {aligned_predictions}")
58
- print(f"Intent logits: {intent_labels}")
59
 
60
  title = "Multi Task Model"
61
  description = '''
 
54
  labels = intent_labels_to_ids()
55
  intent_labels = intent_ids_to_labels(labels)
56
 
57
+ return f"Ner logits: {aligned_predictions}, Intent Label: {intent_labels}"
 
58
 
59
  title = "Multi Task Model"
60
  description = '''