Update app.py

app.py

@@ -54,6 +54,8 @@ def chunk_text_to_window_size_and_predict_proba(input_ids, attention_mask, total
         probabilities = torch.nn.functional.softmax(outputs[0], dim = -1)
         proba_list.append(probabilities)
 
+        start = end
+
 
 
 def get_sentiment(tokens):
@@ -122,6 +124,7 @@ def my_inference_function(sec_txt):
     tokens = tokenizer.encode_plus(txt, add_special_tokens=False)
 
     input_ids = tokens['input_ids']
+    total_len = len(input_ids)
     attention_mask = tokens['attention_mask']
     
     proba_list = chunk_text_to_window_size_and_predict_proba(input_ids, attention_mask, total_len)