Update README.md

README.md

@@ -45,7 +45,7 @@ model.eval()
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
-from transformers import RobertaTokenizer, T5EncoderModel, AutoTokenizer, AutoModelForMaskedLM
+from transformers import T5EncoderModel, AutoTokenizer, AutoModelForMaskedLM
 from huggingface_hub import hf_hub_download
 
 class TemporalFusionClassifier(nn.Module):
@@ -89,7 +89,7 @@ class AICodeDetector:
         self.cb_tokenizer = AutoTokenizer.from_pretrained("microsoft/codebert-base-mlm")
         self.cb_model = AutoModelForMaskedLM.from_pretrained("microsoft/codebert-base-mlm").to(self.device).eval()
 
-        self.t5_tokenizer = RobertaTokenizer.from_pretrained("Salesforce/codet5-base")
+        self.t5_tokenizer = AutoTokenizer.from_pretrained("Salesforce/codet5-base")
         base_t5 = T5EncoderModel.from_pretrained("Salesforce/codet5-base")
         
         weights_path = hf_hub_download(repo_id=repo_id, filename="pytorch_model.bin")
@@ -172,5 +172,5 @@ int main() {
 
 if __name__ == "__main__":
     detector = AICodeDetector()
-    print(detector.analyze(sample))
+    print("\n",detector.analyze(sample))
 ```