Create app.py

app.py

@@ -0,0 +1,38 @@
+import gradio as gr
+
+from transformers import T5ForConditionalGeneration, AutoTokenizer
+from transformers import pipeline
+
+auth_token = os.environ.get("CLARIN_KNEXT")
+
+model_name = "clarin-knext/keywords-plt5-small-shuffle"
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=auth_token)
+model = T5ForConditionalGeneration.from_pretrained(model_name, use_auth_token=auth_token)
+
+
+default_generate_kwargs = {
+            "max_length": 32,
+            "num_beams": 4,
+            "length_penalty": 1.5,
+            "no_repeat_ngram_size": 3,
+            "early_stopping": True,
+        }
+
+keywords_pipe = pipeline(model=model, tokenizer=tokenizer, **default_generate_kwargs)
+
+examples = [
+    ["The Moon's orbit around Earth has"],
+    ["The smooth Borealis basin in the Northern Hemisphere covers 40%"],
+]
+
+def generate(text):
+    return keywords_pipe("keywords: " + text)
+
+demo = gr.Interface(
+    fn=generate,
+    inputs=gr.inputs.Textbox(lines=5, label="Input Text"),
+    outputs=gr.outputs.Textbox(label="Generated Text"),
+    examples=examples
+)
+
+demo.launch()