Updating app to work

app.py

@@ -5,5 +5,39 @@ pipeline = QASRL_Pipeline(model)
 
 description = f"""This is a demo of the '{model}' model, which fine-tuned a Seq2Seq pretrained model on the QANom task""" 
 title="QANom Parser Demo"
-iface = gr.Interface(fn=pipeline, inputs=["text"], outputs=["text"])
+examples = [["The doctor was interested in Luke 's <predicate> treatment .", True, "treat"],
+            ["The Veterinary student was interested in Luke 's <predicate> treatment of sea animals .", True, "treat"]]
+input_sent_box_label = "Insert sentence here. Mark the predicate by adding the token '<p>' before it."
+
+links = """<p style='text-align: center'>
+<a href='https://www.qasrl.org' target='_blank'>QASRL Website</a>
+<a href='https://huggingface.co/spaces/kleinay/qanom-seq2seq-demo' target='_blank'>Model Repo at Huggingface Hub</a> | 
+</p>"""
+
+def call(sentence, is_nominal, verb_form):
+    predicate_marker="<p>"
+    if predicate_marker not in sentence:
+        raise ValueError("You must highlight one word of the sentence as a predicate using preceeding '<p>'.")
+        
+    if not verb_form:
+        if is_nominal:
+            raise ValueError("You should provide the verbal form of the nominalization")
+            
+        toks = sentence.split(" ")
+        pred_idx = toks.index(predicate_marker)
+        predicate = toks(pred_idx+1)
+        verb_form=predicate
+    return pipeline(sentence, 
+                    predicate_marker=predicate_marker, 
+                    predicate_type="nominal" if is_nominal else "verbal",
+                    verb_form=verb_form)
+iface = gr.Interface(fn=call, 
+                     inputs=[gr.inputs.Textbox(placeholder=input_sent_box_label, label="Sentence", lines=4), 
+                             gr.inputs.Checkbox(default=True, label="Is Nominalization?"),
+                             gr.inputs.Textbox(label="verbal form of nominalization", default='')], 
+                     outputs=gr.outputs.JSON(label="Model Output - QASRL"),
+                     title=title,
+                     description=description,
+                     article=links,
+                     examples=examples )
 iface.launch()