Update app.py
Browse files
app.py
CHANGED
|
@@ -69,7 +69,7 @@ def predict(test_query):
|
|
| 69 |
|
| 70 |
from scipy.special import softmax
|
| 71 |
result_arrays_soft = softmax(predict_results[0])
|
| 72 |
-
|
| 73 |
result_array = result_arrays_soft
|
| 74 |
result_list = np.argmax(result_array,axis=-1)
|
| 75 |
asp = []
|
|
|
|
| 69 |
|
| 70 |
from scipy.special import softmax
|
| 71 |
result_arrays_soft = softmax(predict_results[0])
|
| 72 |
+
tag2name = {0: 'O', 1: 'ASP', 2: 'X', 3: '[CLS]', 4: '[SEP]'}
|
| 73 |
result_array = result_arrays_soft
|
| 74 |
result_list = np.argmax(result_array,axis=-1)
|
| 75 |
asp = []
|