Update README.md

README.md

@@ -1,8 +1,8 @@
-Markdown
 ---
 language:
 - en
 - code
+license: other
 tags:
 - pytorch
 - text-classification
@@ -77,6 +77,11 @@ class GraphCodeBERTForClassification(nn.Module):
 base_model = AutoModel.from_pretrained("microsoft/graphcodebert-base")
 tokenizer = AutoTokenizer.from_pretrained("microsoft/graphcodebert-base")
 
+# 3. Initialize the classification model and load weights
+model = GraphCodeBERTForClassification(base_model)
+model.load_state_dict(torch.load("Detect_AI.pth", map_location=torch.device('cpu')))
+model.eval()kenizer = AutoTokenizer.from_pretrained("microsoft/graphcodebert-base")
+
 # 3. Initialize the classification model and load weights
 model = GraphCodeBERTForClassification(base_model)
 model.load_state_dict(torch.load("Detect_AI.pth", map_location=torch.device('cpu')))