Update backup/backup.py
Browse files- backup/backup.py +1 -1
backup/backup.py
CHANGED
|
@@ -12,7 +12,7 @@ def NER_Model(text):
|
|
| 12 |
labels = ["Person", "Mail", "Number", "Address", "Organization","Designation","Link"]
|
| 13 |
|
| 14 |
# Perform entity prediction
|
| 15 |
-
entities = model.predict_entities(text, labels, threshold=0.
|
| 16 |
|
| 17 |
# Initialize the processed data dictionary
|
| 18 |
processed_data = {
|
|
|
|
| 12 |
labels = ["Person", "Mail", "Number", "Address", "Organization","Designation","Link"]
|
| 13 |
|
| 14 |
# Perform entity prediction
|
| 15 |
+
entities = model.predict_entities(text, labels, threshold=0.3)
|
| 16 |
|
| 17 |
# Initialize the processed data dictionary
|
| 18 |
processed_data = {
|