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)
|