kleinay commited on
Commit
5c968f6
·
1 Parent(s): f960349

prettier output

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -26,15 +26,16 @@ def call(sentence, is_nominal, verb_form):
26
  pred_idx = toks.index(predicate_marker)
27
  predicate = toks(pred_idx+1)
28
  verb_form=predicate
29
- return pipeline(sentence,
30
  predicate_marker=predicate_marker,
31
  predicate_type="nominal" if is_nominal else "verbal",
32
  verb_form=verb_form)
 
33
  iface = gr.Interface(fn=call,
34
  inputs=[gr.inputs.Textbox(placeholder=input_sent_box_label, label="Sentence", lines=4),
35
  gr.inputs.Checkbox(default=True, label="Is Nominalization?"),
36
  gr.inputs.Textbox(placeholder=verb_form_inp_placeholder, label="Verbal form (for nominalizations)", default='')],
37
- outputs=gr.outputs.JSON(label="Model Output - QASRL"),
38
  title=title,
39
  description=description,
40
  article=links,
 
26
  pred_idx = toks.index(predicate_marker)
27
  predicate = toks(pred_idx+1)
28
  verb_form=predicate
29
+ pipe_out = pipeline(sentence,
30
  predicate_marker=predicate_marker,
31
  predicate_type="nominal" if is_nominal else "verbal",
32
  verb_form=verb_form)
33
+ return pipe_out[0]["QAs"], pipe_out[0]["generated_text"]
34
  iface = gr.Interface(fn=call,
35
  inputs=[gr.inputs.Textbox(placeholder=input_sent_box_label, label="Sentence", lines=4),
36
  gr.inputs.Checkbox(default=True, label="Is Nominalization?"),
37
  gr.inputs.Textbox(placeholder=verb_form_inp_placeholder, label="Verbal form (for nominalizations)", default='')],
38
+ outputs=[gr.outputs.JSON(label="Model Output - QASRL"), gr.outputs.Textbox(label="Raw output sequence")],
39
  title=title,
40
  description=description,
41
  article=links,