Update app.py

app.py

@@ -1,8 +1,35 @@
 import gradio as gr
+from transformers import pipeline
+from gradio_client import Client
 
-def saluda(nom):
-  return "Hola " + nom
+client = Client("abidlabs/whisper")
 
-demo = gr.Interface(fn=saluda, inputs="text", outputs="text")
+ner_pipeline = pipeline(model="projecte-aina/roberta-base-ca-v2-cased-ner")
 
-demo.launch(auth=("admin", "1234"))
+def ner(text):
+    output = ner_pipeline(text)
+    return {"text": text, "entities": output}
+    
+def neteja():
+    return [gr.Textbox(value=None), gr.HighlightedText(value=None)]
+
+with gr.Blocks(theme=gr.themes.Glass()) as demo:
+    gr.Markdown(
+    """
+    # Reconeixament d'entitats nomenades en català
+    Escriu o copia un text i troba les seves entitats
+    """)
+    with gr.Row():
+      with gr.Column():
+        inp = gr.Textbox(label="text", placeholder="Escriu aquí...")
+        with gr.Row():
+          b1 = gr.Button(value="Neteja")
+          b2 = gr.Button("Troba entitats", variant="primary")
+      out = gr.HighlightedText(label="sortida")
+
+    examples = gr.Examples(examples=exemples, inputs=inp, label="Exemple:")
+
+    b1.click(neteja, outputs=[inp, out])
+    b2.click(ner, inputs=inp, outputs=out)
+
+demo.launch(auth=("admin", "admin"))