question answering text

Pipfile

@@ -1,14 +0,0 @@
-[[source]]
-url = "https://pypi.org/simple"
-verify_ssl = true
-name = "pypi"
-
-[packages]
-transformers = "*"
-gradio = "*"
-
-[dev-packages]
-
-[requires]
-python_version = "3.9"
-python_full_version = "3.9.13"

README.md

@@ -1,5 +1,5 @@
 ---
-title: Vacancy Keyword Extraction
+title: Question Answer Text
 emoji: 🚀
 colorFrom: blue
 colorTo: white

app.py

@@ -1,17 +1,64 @@
 import gradio as gr
-from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
-title = "Vacancy Keyword Extraction"
-description = "Add a vacancy and ask what keyword you would like to retrieve. For example, What is the job title? What is the city? What is the start date?"
+from transformers import pipeline
 
+title = "Question Answer on Text"
+description = "Provide a piece of text, ask a question and get a meaningful answer."
 
-classifier = pipeline("question-answering", model="deepset/roberta-base-squad2", model_max_length=512)
 
-def zeroShotClassification(text_input, question):
-    prediction = classifier(context=text_input, question=question, truncation=True, max_length="max_length", padding=True)
+classifier = pipeline(
+    "question-answering", model="deepset/roberta-base-squad2", model_max_length=512
+)
+
+
+def zero_shot_classification(text_input, question):
+    prediction = classifier(
+        context=text_input,
+        question=question,
+        truncation=True,
+        max_length="max_length",
+        padding=True,
+    )
     return f'{prediction["answer"]}'
 
-gr.Interface(fn=zeroShotClassification,
-                  inputs=[gr.inputs.Textbox(lines=10, label="Vacancy", placeholder="Paste the context of a vacancy."),
-                          gr.inputs.Textbox(lines=2, label="Question", placeholder="Ask what you want to retrieve from the vacancy.")],
-                  outputs=gr.outputs.Textbox(label="Answer"),
-                  title=title, description=description).launch(share=False)
+
+examples = [
+    [
+        """The invoice contains all the goods that are described in the order with number x9820be. 
+    The goods are packaged and should arrive at the destination on 31/12/2023. 
+    We kindly request to transfer the amount within 30 days of the invoice date to the account number BE00 9999 9999 9999""",
+        "what is the timeline to transfer the money?",
+    ],
+    [
+        """The invoice contains all the goods that are described in the order with number x9820be. 
+    The goods are packaged and should arrive at the destination on 31/12/2023. 
+    We kindly request to transfer the amount within 30 days of the invoice date to the account number BE00 9999 9999 9999""",
+        "what is the account number to transfer the money?",
+    ],[
+        """The invoice contains all the goods that are described in the order with number x9820be. 
+    The goods are packaged and should arrive at the destination on 31/12/2023. 
+    We kindly request to transfer the amount within 30 days of the invoice date to the account number BE00 9999 9999 9999""",
+        "order number",
+    ],
+    [
+        """By signing below, Shipper hereby declares that the contents of this consignment are fully and accurately described above by the proper shipping name and are classified,
+packaged, marked and labelled/placarded, and are in all respects in proper condition for transport according to applicable governmental regulations. As shipper, I hereby
+certify that the liquid industrial by-product(s) are fully and accurately described on this shipping document, in proper condition for transport, and that the information
+contained on the shipping document is factual.""",
+        "are the goods described?",
+    ],
+]
+
+gr.Interface(
+    fn=zero_shot_classification,
+    inputs=[
+        gr.inputs.Textbox(lines=10, label="Text", placeholder="Paste text here ..."),
+        gr.inputs.Textbox(
+            lines=2,
+            label="Question",
+            placeholder="Ask what you want to retrieve from the vacancy.",
+        ),
+    ],
+    outputs=gr.outputs.Textbox(label="Answer"),
+    title=title,
+    description=description,
+).launch(share=False)