yangheng commited on
Commit
ce4ab3b
1 Parent(s): 958f16d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)