Refactor parse_files

app.py

@@ -9,7 +9,8 @@ tokenizer = RobertaTokenizer.from_pretrained("mamiksik/CommitPredictorT5PL", rev
 model = T5ForConditionalGeneration.from_pretrained("mamiksik/CommitPredictorT5PL", revision="fb08d01")
 
 
-def parse_files(accumulator, patch):
+def parse_files(patch):
+    accumulator = []
     lines = patch.splitlines()
 
     filename_before = None
@@ -43,14 +44,11 @@ def parse_files(accumulator, patch):
 
         accumulator.append(line)
 
-    return accumulator
+    return '\n'.join(accumulator)
 
 
 def predict(patch, max_length, min_length, num_beams, prediction_count):
-    accumulator = []
-    parse_files(accumulator, patch)
-    input_text = '\n'.join(accumulator)
-
+    input_text = parse_files(patch)
     with torch.no_grad():
         token_count = tokenizer(input_text, return_tensors="pt").input_ids.shape[1]