Spaces:
Sleeping
Sleeping
Commit
Β·
419de15
1
Parent(s):
774d000
Fix bug
Browse files
pages/3_π_Zero Shot Classification.py
CHANGED
@@ -53,10 +53,12 @@ if st.button("Classify Text"):
|
|
53 |
st.write("## Classification Results:")
|
54 |
probabilities = []
|
55 |
result = zero_shot(user_input, labels)
|
56 |
-
|
|
|
57 |
for label, score in zip(result['labels'], result['scores']):
|
|
|
58 |
probabilities.append(round(score, 2))
|
59 |
-
|
60 |
fig = go.Figure(data=[
|
61 |
go.Bar(
|
62 |
x=labels,
|
|
|
53 |
st.write("## Classification Results:")
|
54 |
probabilities = []
|
55 |
result = zero_shot(user_input, labels)
|
56 |
+
|
57 |
+
labels.clear()
|
58 |
for label, score in zip(result['labels'], result['scores']):
|
59 |
+
labels.append(label)
|
60 |
probabilities.append(round(score, 2))
|
61 |
+
|
62 |
fig = go.Figure(data=[
|
63 |
go.Bar(
|
64 |
x=labels,
|