Spaces:
Sleeping
Sleeping
fixed the predict function
Browse files
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 |
-
|
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 = '''
|