File size: 276 Bytes
1b36ad4
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
import gradio as gr
from transformers import pipeline

ner = pipeline(
    "NER-indian",
    model="techysanoj/fine-tuned-IndicNER",
    aggregation_strategy="simple"
)

def predict(text):
    return ner(text)

gr.Interface(fn=predict, inputs="text", outputs="json").launch()