Spaces:
Runtime error
Runtime error
Jainesh212
commited on
Commit
•
214f718
1
Parent(s):
ede77af
Update app.py
Browse files
app.py
CHANGED
@@ -4,19 +4,19 @@ import matplotlib.pyplot as plt
|
|
4 |
import seaborn as sns
|
5 |
import transformers
|
6 |
|
7 |
-
|
8 |
models = {
|
9 |
"DistilBERT": transformers.pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english"),
|
10 |
"RoBERTa": transformers.pipeline("sentiment-analysis", model="roberta-base-openai-detector"),
|
11 |
}
|
12 |
|
13 |
-
|
14 |
def analyze_sentiment(text, model_name):
|
15 |
model = models[model_name]
|
16 |
result = model(text)[0]
|
17 |
return result['label'], result['score']
|
18 |
|
19 |
-
|
20 |
def app():
|
21 |
st.title("Sentiment Analysis App")
|
22 |
|
@@ -43,6 +43,6 @@ def app():
|
|
43 |
ax.set_title("Sentiment Analysis Results")
|
44 |
st.pyplot(fig)
|
45 |
|
46 |
-
|
47 |
if __name__ == "__main__":
|
48 |
app()
|
|
|
4 |
import seaborn as sns
|
5 |
import transformers
|
6 |
|
7 |
+
|
8 |
models = {
|
9 |
"DistilBERT": transformers.pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english"),
|
10 |
"RoBERTa": transformers.pipeline("sentiment-analysis", model="roberta-base-openai-detector"),
|
11 |
}
|
12 |
|
13 |
+
|
14 |
def analyze_sentiment(text, model_name):
|
15 |
model = models[model_name]
|
16 |
result = model(text)[0]
|
17 |
return result['label'], result['score']
|
18 |
|
19 |
+
|
20 |
def app():
|
21 |
st.title("Sentiment Analysis App")
|
22 |
|
|
|
43 |
ax.set_title("Sentiment Analysis Results")
|
44 |
st.pyplot(fig)
|
45 |
|
46 |
+
|
47 |
if __name__ == "__main__":
|
48 |
app()
|