Fixed labels
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ model = AutoModelForTokenClassification.from_pretrained(model_directory)
|
|
| 12 |
model.eval()
|
| 13 |
|
| 14 |
# Define the label map
|
| 15 |
-
label_map = {0: "O", 1: "B-INDICATOR", 2: "I-INDICATOR", 3: "B-CAUSE", 4: "I-CAUSE"}
|
| 16 |
|
| 17 |
# Streamlit App
|
| 18 |
st.markdown(
|
|
|
|
| 12 |
model.eval()
|
| 13 |
|
| 14 |
# Define the label map
|
| 15 |
+
label_map = {0: "O", 1: "B-INDICATOR", 2: "I-INDICATOR", 3: "B-CAUSE", 4: "I-CAUSE", 5: "B-EFFECT", 6: "I-EFFECT"}
|
| 16 |
|
| 17 |
# Streamlit App
|
| 18 |
st.markdown(
|