Update app.py

app.py

@@ -2,10 +2,10 @@ from transformers import pipeline
 
 import gradio as gr
 
-pretrained_name = "cahya/bert-base-indonesian-NER"
+pretrained_name = "w11wo/indonesian-roberta-base-sentiment-classifier"
 
-ner = pipeline(
-    "ner",
+sentiment = pipeline(
+    "sentiment-analysis",
     model=pretrained_name,
     tokenizer=pretrained_name
 )
@@ -14,14 +14,17 @@ examples = [
     "Ada apa dengan Jokowi di istana negara?",
 ]
 
-def ner(text):
-    output = ner(text)
-    return {"text": text, "entities": output}    
+def sentiment_analysis(text):
+    scores = sid.polarity_scores(text)
+    del scores["compound"]
+    return scores 
 
-demo = gr.Interface(ner,
-             gr.Textbox(placeholder="Enter sentence here..."), 
-             gr.HighlightedText(),
-             examples=examples)
+demo = gr.Interface(
+    fn=sentiment_analysis, 
+    inputs=gr.Textbox(placeholder="Enter a sentence here..."), 
+    outputs="label", 
+    interpretation="default",
+    examples=[examples])
 
 if __name__ == "__main__":
     demo.launch()