Update app.py
Browse files
app.py
CHANGED
|
@@ -165,7 +165,7 @@ def perform_acos_inference(text, dataset):
|
|
| 165 |
random.randint(0, len(acos_dataset_dict[dataset]) - 1)
|
| 166 |
]
|
| 167 |
|
| 168 |
-
raw_output = quadruple_extractor.predict(text.split('####')[0])
|
| 169 |
|
| 170 |
result = raw_output['Quadruples']
|
| 171 |
result = pd.DataFrame(result)
|
|
|
|
| 165 |
random.randint(0, len(acos_dataset_dict[dataset]) - 1)
|
| 166 |
]
|
| 167 |
|
| 168 |
+
raw_output = quadruple_extractor.predict(text.split('####')[0], max_length=128)
|
| 169 |
|
| 170 |
result = raw_output['Quadruples']
|
| 171 |
result = pd.DataFrame(result)
|