Update README.md

README.md

@@ -1,3 +1,28 @@
----
-license: cc-by-nc-3.0
----
+---
+license: cc-by-nc-3.0
+base_model:
+- clue/albert_chinese_tiny
+---
+
+
+```python
+from transformers import AlbertForSequenceClassification, AutoTokenizer
+import torch
+
+# 加载模型和分词器
+model_name = "./results/checkpoint-3118"  # 中文ALBERT-Tiny（仅18MB）
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AlbertForSequenceClassification.from_pretrained(model_name, num_labels=2)
+
+# print(model)
+def predict(text):
+    inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=64)
+    with torch.no_grad():
+        outputs = model(**inputs)
+        print(outputs)
+    return torch.argmax(outputs.logits).item()
+
+text = "1、消失的他"
+
+pred = predict(text)
+```