HamidBekam commited on
Commit
b63a2e3
1 Parent(s): 80d7e63

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -0
app.py ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # app.py
2
+ import gradio as gr
3
+ from transformers import pipeline
4
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
5
+
6
+ #Defining the classify function which takes text as input and returns the label of the sentiment
7
+ def classify(text):
8
+ # Initializing the pipeline for sentiment analysis
9
+ cls = pipeline('text-classification', model='RJuro/dk_emotion_bert_in_class')
10
+ # Predicting the sentiment label for the input text
11
+ return cls(text)[0]['label']
12
+
13
+ #Creating the Gradio interface with input textbox and output text
14
+ gr.Interface(fn=classify, inputs=["textbox"], outputs="text").launch()