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 = '''
|