Himanshusingh commited on
Commit
e7cba8e
1 Parent(s): 567a866

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def chunk_text_to_window_size_and_predict_proba(input_ids, attention_mask, total
49
  while loop:
50
  end = start + window_length
51
  # If the end index exceeds total length, set the flag to False and adjust the end index
52
- if end >= total_len:
53
  loop = False
54
  end = total_len
55
 
 
49
  while loop:
50
  end = start + window_length
51
  # If the end index exceeds total length, set the flag to False and adjust the end index
52
+ if end >= total_len OR count == 25:
53
  loop = False
54
  end = total_len
55