Update input

.gitignore

@@ -0,0 +1 @@
+__pycache__

app.py

@@ -1,15 +1,31 @@
 import gradio as gr
 from transformers import pipeline
 
-# pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
+summarizer = pipeline(task="summarization", model="facebook/bart-large-cnn")
 
-def predict(name, intensity):
-    return "Hello, " + name + "!" * int(intensity)
+def fetch_record(record_url):
+    return "https://images.unsplash.com/photo-1507003211169-0a1dd7228f2d?ixlib=rb-1.2.1&ixid=MnwxMjA3fDB8MHxwaG90by1wYWdlfHx8fGVufDB8fHx8&auto=format&fit=crop&w=387&q=80"
 
-app = gr.Interface(
-    fn=predict,
-    inputs=["text", "slider"],
-    outputs=["text"],
-)
+def summary_text(input_text: str, min_length: int = 30, max_length: int = 120):
+    summary = summarizer(input_text, max_length=max_length, min_length=min_length, do_sample=False)
+    return summary[0]['summary_text']
 
-app.launch(server_name='0.0.0.0')
+with gr.Blocks() as app:
+    gr.Markdown("# SLNSW - Text sumarisation from the input above")
+    gr.Markdown("Summarise long text description either by record URL or manual copy-and-paste.")
+    with gr.Row():
+        with gr.Column():
+            record_url = gr.Text(label="SL record URL", lines=5, info="The main URL of collections eg. https://collection.sl.nsw.gov.au/record/YdmdvKj9")
+            fetch_record_btn = gr.Button("Fetch URL")
+        with gr.Column():
+            input_text = gr.Text(label="Input text", lines=5, info="The manual input of long text", interactive=True)
+            summary_text_btn = gr.Button("Summary")
+    with gr.Row():
+        with gr.Column():
+            summary_text_txt = gr.Text(label="Summary", info="The summarised text")
+    
+    fetch_record_btn.click(fn=fetch_record, inputs=[record_url], outputs=[input_text])
+    summary_text_btn.click(fn=summary_text, inputs=[input_text], outputs=[summary_text_txt], api_name="summary_text")
+
+if __name__ == "__main__":
+    app.launch(server_name='0.0.0.0', ssl_verify=False)

taskfile.yaml

@@ -1,19 +0,0 @@
-version: "3"
-env:
-  NAME: "slnsw-text-summary"
-tasks:
-  setup-hf:
-    cmds:
-      - huggingface-cli repo create --type model -y $NAME
-      - huggingface-cli repo create --type space --space_sdk gradio -y $NAME
-      - huggingface-cli repo create --type dataset -y $NAME
-  setup-git:
-    cmds:
-      - git remote add hf-model https://huggingface.co/generaptor/$NAME
-      - git remote add hf-space https://huggingface.co/spaces/generaptor/$NAME
-      - git clone https://huggingface.co/datasets/generaptor/$NAME datasets
-      - git submodule add ./datasets hf-datasets
-  setup:
-    cmds:
-      - task: "setup-hf"
-      - task: "setup-git"