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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -167,7 +167,7 @@ def perform_acos_inference(text, dataset):
167
 
168
  raw_output = quadruple_extractor.predict(text.split('####')[0])
169
 
170
- result = raw_output['Quadruple']
171
  result = pd.DataFrame(result)
172
  return result, text
173
 
@@ -188,7 +188,7 @@ with demo:
188
  )
189
  acos_dataset_ids = gr.Radio(
190
  choices=[dataset.name for dataset in ABSAInstruction.ACOSDatasetList()],
191
- value="Restaurant16",
192
  label="Datasets",
193
  )
194
  acos_inference_button = gr.Button("Let's go!")
@@ -269,4 +269,4 @@ with demo:
269
  """
270
  )
271
 
272
- demo.launch()
 
167
 
168
  raw_output = quadruple_extractor.predict(text.split('####')[0])
169
 
170
+ result = raw_output['Quadruples']
171
  result = pd.DataFrame(result)
172
  return result, text
173
 
 
188
  )
189
  acos_dataset_ids = gr.Radio(
190
  choices=[dataset.name for dataset in ABSAInstruction.ACOSDatasetList()],
191
+ value="Laptop14",
192
  label="Datasets",
193
  )
194
  acos_inference_button = gr.Button("Let's go!")
 
269
  """
270
  )
271
 
272
+ demo.launch()