bug fixes

app.py

@@ -1,9 +1,9 @@
 import re
 
 import gradio as gr
-from transformers import RobertaTokenizer, pipeline, AutoModelForMaskedLM
+from transformers import pipeline, AutoModelForMaskedLM, AutoTokenizer
 
-tokenizer = RobertaTokenizer.from_pretrained("mamiksik/CodeBERTa-commit-message-autocomplete")
+tokenizer = AutoTokenizer.from_pretrained("mamiksik/CodeBERTa-commit-message-autocomplete")
 model = AutoModelForMaskedLM.from_pretrained("mamiksik/CodeBERTa-commit-message-autocomplete")
 pipe = pipeline("fill-mask", model=model, tokenizer=tokenizer)
 
@@ -47,7 +47,7 @@ def parse_files(patch) -> str:
 
 
 def predict(patch, commit_message):
-    input_text = parse_files(patch) + "\n<msg> " + commit_message
+    input_text = parse_files(patch) + "</sep></sep><msg> " + commit_message + "</sep>"
     token_count = tokenizer(input_text, return_tensors="pt").input_ids.shape[1]
     result = pipe.predict(input_text)