File size: 646 Bytes
34bf412
6e342ef
 
0539c21
6e342ef
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
34bf412
 
 
6e342ef
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
import gradio as gr
from transformers import pipeline

pipeline = pipeline(task="text-classification", model="Preetham04/sentiment_analysis")

def predict(input_text):
    predictions = pipeline(input_text)
    print(predictions)
    result = [] 
    for p in predictions:
        review = "POSITIVE" if p["label"] == "LABEL_1" else "NEGATIVE"
        result.append({
            "SENTIMENT": review,
            "SCORE": p["score"]
        })
    return result

gradio_app = gr.Interface(
    predict,
    inputs="textbox",
    outputs="text",
    title="Sentiment- good or bad?",
)

if __name__ == "__main__":
    gradio_app.launch(share=True)