Added ViT Model

README.md

@@ -13,17 +13,24 @@ datasets:
 model_name: BertAndDeberta
 ---
 
-# BertAndDeberta — Transformer Ensemble for Fake News Detection
+# BertAndDeberta + ViT — Transformer Ensemble for Fake News Detection
 
-This repository hosts a fine-tuned transformer ensemble model that combines **BERT** and **DeBERTa** for binary **fake news classification**. The model was trained on a custom-labeled dataset containing real and fake news articles sourced from Kaggle and Hugging Face datasets.
+This repository hosts a multimodal fake news detection system that combines **BERT**, **DeBERTa**, and **ViT** models. The *BERT* and *DeBERTa* models handle textual data to classify news as either real or fake, while the *ViT* model is trained separately to detect AI-generated vs real images, helping assess the authenticity of visual content.
 
-## Model Details
 
+##  Text Models — Fake News Detection
 - **Architecture**: Ensemble of BERT-base and DeBERTa-base
 - **Task**: Binary Text Classification (`REAL` vs `FAKE`)
 - **Training Framework**: PyTorch using 🤗 Transformers
 - **License**: MIT
 
+    
+## Vision Model — AI-Generated Image Detection
+- **Architecture**: Vision Transformer (ViT-base, vit-base-patch16-224)
+- **Task**: Binary Image Classification ('REAL' vs 'AI-GENERATED')
+- **Training Framework**: TensorFlow/Keras or PyTorch using Transformers
+- **License**: MIT
+
 ## Dataset
 
 The dataset is a custom collection combining: