Spaces:
Build error
Build error
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,
|