Update README.md

README.md

@@ -28,6 +28,7 @@ license: mit
 
 ## Usage Example
 
+Direct Model Usage
 ```python
 # Install semviqa
 !pip install semviqa
@@ -37,11 +38,11 @@ from transformers import AutoTokenizer
 from semviqa.ser.qatc_model import QATCForQuestionAnswering
 import torch
 
-tokenizer = AutoTokenizer.from_pretrained("xuandin/semviqa-qatc-vi-mrc-large-viwikifc")
-model = QATCForQuestionAnswering.from_pretrained("xuandin/semviqa-qatc-vi-mrc-large-viwikifc")
+tokenizer = AutoTokenizer.from_pretrained("SemViQA/qatc-vimrc-viwikifc")
+model = QATCForQuestionAnswering.from_pretrained("SemViQA/qatc-vimrc-viwikifc")
 claim = "Chiến tranh với Campuchia đã kết thúc trước khi Việt Nam thống nhất."
 context = "Sau khi thống nhất, Việt Nam tiếp tục gặp khó khăn do sự sụp đổ và tan rã của đồng minh Liên Xô cùng Khối phía Đông, các lệnh cấm vận của Hoa Kỳ, chiến tranh với Campuchia, biên giới giáp Trung Quốc và hậu quả của chính sách bao cấp sau nhiều năm áp dụng. Năm 1986, Đảng Cộng sản ban hành cải cách đổi mới, tạo điều kiện hình thành kinh tế thị trường và hội nhập sâu rộng. Cải cách đổi mới kết hợp cùng quy mô dân số lớn đưa Việt Nam trở thành một trong những nước đang phát triển có tốc độ tăng trưởng thuộc nhóm nhanh nhất thế giới, được coi là Hổ mới châu Á dù cho vẫn gặp phải những thách thức như tham nhũng, tội phạm gia tăng, ô nhiễm môi trường và phúc lợi xã hội chưa đầy đủ. Ngoài ra, giới bất đồng chính kiến, chính phủ một số nước phương Tây và các tổ chức theo dõi nhân quyền có quan điểm chỉ trích hồ sơ nhân quyền của Việt Nam liên quan đến các vấn đề tôn giáo, kiểm duyệt truyền thông, hạn chế hoạt động ủng hộ nhân quyền cùng các quyền tự do dân sự."
- 
+
 inputs = tokenizer(claim, context, return_tensors="pt", truncation=True, max_length=512)
 
 with torch.no_grad():
@@ -59,6 +60,28 @@ print(evidence)
 # evidence: Sau khi thống nhất, Việt Nam tiếp tục gặp khó khăn do sự sụp đổ và tan rã của đồng minh Liên Xô cùng Khối phía Đông, các lệnh cấm vận của Hoa Kỳ, chiến tranh với Campuchia, biên giới giáp Trung Quốc và hậu quả của chính sách bao cấp sau nhiều năm áp dụng.
 ```
 
+Using TF-IDF and QATC Combination with Confidence Threshold
+```python
+import torch
+from transformers import AutoTokenizer
+from semviqa.ser.qatc_model import QATCForQuestionAnswering
+from semviqa.ser.ser_eval import extract_evidence_tfidf_qatc
+
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
+tokenizer = AutoTokenizer.from_pretrained("SemViQA/qatc-vimrc-viwikifc")
+model = QATCForQuestionAnswering.from_pretrained("SemViQA/qatc-vimrc-viwikifc").to(device)
+
+claim = "Chiến tranh với Campuchia đã kết thúc trước khi Việt Nam thống nhất."
+context = "Sau khi thống nhất, Việt Nam tiếp tục gặp khó khăn do sự sụp đổ và tan rã của đồng minh Liên Xô cùng Khối phía Đông, các lệnh cấm vận của Hoa Kỳ, chiến tranh với Campuchia, biên giới giáp Trung Quốc và hậu quả của chính sách bao cấp sau nhiều năm áp dụng. Năm 1986, Đảng Cộng sản ban hành cải cách đổi mới, tạo điều kiện hình thành kinh tế thị trường và hội nhập sâu rộng. Cải cách đổi mới kết hợp cùng quy mô dân số lớn đưa Việt Nam trở thành một trong những nước đang phát triển có tốc độ tăng trưởng thuộc nhóm nhanh nhất thế giới, được coi là Hổ mới châu Á dù cho vẫn gặp phải những thách thức như tham nhũng, tội phạm gia tăng, ô nhiễm môi trường và phúc lợi xã hội chưa đầy đủ. Ngoài ra, giới bất đồng chính kiến, chính phủ một số nước phương Tây và các tổ chức theo dõi nhân quyền có quan điểm chỉ trích hồ sơ nhân quyền của Việt Nam liên quan đến các vấn đề tôn giáo, kiểm duyệt truyền thông, hạn chế hoạt động ủng hộ nhân quyền cùng các quyền tự do dân sự."
+
+evidence = extract_evidence_tfidf_qatc(
+    claim, context, model, tokenizer, device, confidence_threshold=0.5, length_ratio_threshold=0.6
+)
+
+print(evidence)
+# evidence: sau khi thống nhất việt nam tiếp tục gặp khó khăn do sự sụp đổ và tan rã của đồng minh liên xô cùng khối phía đông các lệnh cấm vận của hoa kỳ chiến tranh với campuchia biên giới giáp trung quốc và hậu quả của chính sách bao cấp sau nhiều năm áp dụng
+```
 ---
 
 ## **Evaluation Results**