Update app.py
Browse files
app.py
CHANGED
|
@@ -5,9 +5,9 @@ classifier = pipeline("text-classification", model="sergiopaniego/modernbert-ban
|
|
| 5 |
|
| 6 |
def get_text_classification(text_input):
|
| 7 |
prediction = classifier(text_input)
|
| 8 |
-
|
| 9 |
-
|
| 10 |
-
return
|
| 11 |
|
| 12 |
examples = [["When did you send me my new card?"]]
|
| 13 |
|
|
|
|
| 5 |
|
| 6 |
def get_text_classification(text_input):
|
| 7 |
prediction = classifier(text_input)
|
| 8 |
+
formatted_text = f"{prediction[0]['label']}: {prediction[0]['score']}"
|
| 9 |
+
|
| 10 |
+
return formatted_text
|
| 11 |
|
| 12 |
examples = [["When did you send me my new card?"]]
|
| 13 |
|