Himanshusingh
commited on
Commit
•
ffe7a35
1
Parent(s):
e7cba8e
Update app.py
Browse files
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 |
|