ahmadtalha commited on
Commit
419de15
Β·
1 Parent(s): 774d000
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,