kusa04 commited on
Commit
88d993b
·
verified ·
1 Parent(s): 012fb93

Update functions.py

Browse files
Files changed (1) hide show
  1. functions.py +7 -1
functions.py CHANGED
@@ -68,14 +68,20 @@ def split_text_by_token_limit(text, tokenizer, max_tokens):
68
  return chunks
69
 
70
 
71
- def safe_sentiment(sentiment_pipeline, text):
 
 
72
  try:
73
  result = sentiment_pipeline(text)[0]
74
  except Exception as e:
75
  result = None
 
 
 
76
  return result
77
 
78
 
 
79
  def safe_sentiment_batch(sentiment_pipeline, texts):
80
  try:
81
  results = sentiment_pipeline(texts)
 
68
  return chunks
69
 
70
 
71
+ def safe_sentiment(sentiment_pipeline, text, length):
72
+ if "count" not in st.session_state:
73
+ st.session_state.count = 0
74
  try:
75
  result = sentiment_pipeline(text)[0]
76
  except Exception as e:
77
  result = None
78
+
79
+ st.session_state.count += 1
80
+ st.write(f"Phase: {st.session_state.count / length}")
81
  return result
82
 
83
 
84
+
85
  def safe_sentiment_batch(sentiment_pipeline, texts):
86
  try:
87
  results = sentiment_pipeline(texts)