Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -7,4 +7,11 @@ inputs = [gr.Textbox(value = "I love this!")]
|
|
7 |
outputs = [gr.Label(label = "Sentiment")]
|
8 |
title = "Sentiment Analysis Classifier"
|
9 |
description = "This is a sentiment classifier using longformer model with a logistic regression head. "
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
10 |
gr.Interface(infer, inputs = inputs, outputs = outputs, title = title, description = description).launch()
|
|
|
7 |
outputs = [gr.Label(label = "Sentiment")]
|
8 |
title = "Sentiment Analysis Classifier"
|
9 |
description = "This is a sentiment classifier using longformer model with a logistic regression head. "
|
10 |
+
def infer(inputs):
|
11 |
+
predictions = pipe.predict_proba([inputs])
|
12 |
+
label = {
|
13 |
+
"negative":str(predictions[0][0]),
|
14 |
+
"positive":str(predictions[0][1]),
|
15 |
+
}
|
16 |
+
return label
|
17 |
gr.Interface(infer, inputs = inputs, outputs = outputs, title = title, description = description).launch()
|