Spaces:
Runtime error
Runtime error
updated categroies on the fly
Browse files
app.py
CHANGED
|
@@ -27,9 +27,12 @@ db_cursor = db_connection.cursor()
|
|
| 27 |
|
| 28 |
ORG_ID = 731
|
| 29 |
|
|
|
|
|
|
|
| 30 |
|
| 31 |
def get_potential_labels():
|
| 32 |
# get potential labels from db
|
|
|
|
| 33 |
potential_labels = db_cursor.execute(
|
| 34 |
"SELECT message_category_name FROM radmap_frog12.message_categorys"
|
| 35 |
)
|
|
@@ -46,6 +49,7 @@ potential_labels = get_potential_labels()
|
|
| 46 |
|
| 47 |
# Function to handle the classification
|
| 48 |
def classify_email(constituent_email):
|
|
|
|
| 49 |
print("classifying email")
|
| 50 |
model_out = classifier_model(constituent_email, potential_labels, multi_label=True)
|
| 51 |
print("classification complete")
|
|
|
|
| 27 |
|
| 28 |
ORG_ID = 731
|
| 29 |
|
| 30 |
+
potential_labels = []
|
| 31 |
+
|
| 32 |
|
| 33 |
def get_potential_labels():
|
| 34 |
# get potential labels from db
|
| 35 |
+
global potential_labels
|
| 36 |
potential_labels = db_cursor.execute(
|
| 37 |
"SELECT message_category_name FROM radmap_frog12.message_categorys"
|
| 38 |
)
|
|
|
|
| 49 |
|
| 50 |
# Function to handle the classification
|
| 51 |
def classify_email(constituent_email):
|
| 52 |
+
potential_labels = get_potential_labels()
|
| 53 |
print("classifying email")
|
| 54 |
model_out = classifier_model(constituent_email, potential_labels, multi_label=True)
|
| 55 |
print("classification complete")
|