kyled commited on
Commit
31cc01d
1 Parent(s): 4d85e89

added tensorflow

Browse files
Files changed (2) hide show
  1. app.py +20 -1
  2. requirements.txt +2 -0
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import streamlit as st
2
  import transformers as t
 
3
 
4
  st.title("Phrase Feeling Analysis")
5
  classifier = t.pipeline("zero-shot-classification",
@@ -10,4 +11,22 @@ x = st.text_input("Enter your title here:")
10
  candidate_labels = ['anger', 'sadness', 'fear', 'joy', 'interest',
11
  'surprise', 'disgust', 'shame', 'guilt', 'compassion', 'other']
12
  output = classifier(x, candidate_labels)
13
- st.write(output)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import streamlit as st
2
  import transformers as t
3
+ import plotly.graph_objects as go
4
 
5
  st.title("Phrase Feeling Analysis")
6
  classifier = t.pipeline("zero-shot-classification",
 
11
  candidate_labels = ['anger', 'sadness', 'fear', 'joy', 'interest',
12
  'surprise', 'disgust', 'shame', 'guilt', 'compassion', 'other']
13
  output = classifier(x, candidate_labels)
14
+ # st.write(output)
15
+
16
+ for i in range(len(candidate_labels)):
17
+ st.write(candidate_labels[i] + ": " + str(output[i]))
18
+
19
+ # fig = go.Figure(data=go.Scatterpolar(
20
+ # r=[1, 5, 2, 2, 3],
21
+ # theta=candidate_labels,
22
+ # fill='toself'
23
+ # ))
24
+
25
+ # fig.update_layout(
26
+ # polar=dict(
27
+ # radialaxis=dict(
28
+ # visible=True
29
+ # ),
30
+ # ),
31
+ # showlegend=False
32
+ # )
requirements.txt CHANGED
@@ -2,3 +2,5 @@ streamlit
2
  transformers
3
  pandas
4
  plotly
 
 
 
2
  transformers
3
  pandas
4
  plotly
5
+ tensorflow
6
+ pytorch