Spaces:
Sleeping
Sleeping
Update app_11.py
Browse files
app_11.py
CHANGED
|
@@ -85,12 +85,6 @@ def extract_arguments(text, tokenizer, model, beam_search=True):
|
|
| 85 |
start_effect_logits[len(inputs["input_ids"][0]) - 1] = -1e-4
|
| 86 |
end_effect_logits[len(inputs["input_ids"][0]) - 1] = -1e-4
|
| 87 |
|
| 88 |
-
st.write("start_cause_logits", start_cause_logits)
|
| 89 |
-
st.write("end_cause_logits", end_cause_logits)
|
| 90 |
-
st.write("start_effect_logits", start_effect_logits)
|
| 91 |
-
st.write("end_effect_logits", end_effect_logits)
|
| 92 |
-
st.write("start_signal_logits", start_signal_logits)
|
| 93 |
-
st.write("end_signal_logits", end_signal_logits)
|
| 94 |
|
| 95 |
# Beam Search for position selection
|
| 96 |
if beam_search:
|
|
|
|
| 85 |
start_effect_logits[len(inputs["input_ids"][0]) - 1] = -1e-4
|
| 86 |
end_effect_logits[len(inputs["input_ids"][0]) - 1] = -1e-4
|
| 87 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 88 |
|
| 89 |
# Beam Search for position selection
|
| 90 |
if beam_search:
|