changed config and readme

README.md

@@ -1,25 +1,17 @@
 ---
-language: 
-- en
-license: mit
+language: en
+library_name: pytorch
+license: apache-2.0
 tags:
+- image-classification
 - deepfake-detection
-- vision-transformer
-- efficientnet
-- multimodal
+- efficientvit
 - pytorch
-- computer-vision
-- model
-- image-classification
-datasets:
-- custom
-metrics:
-- accuracy
-- f1
 pipeline_tag: image-classification
-library_name: pytorch
-widget:
-- text: "sample_video.mp4"
+model-index:
+- name: Deepfake Detection with Improved EfficientViT
+  results: []
+  config: config.json
 ---
 
 # Deepfake Detection with Improved EfficientViT

config.json

@@ -1,14 +1,16 @@
 {
+  "architectures": ["ImprovedEfficientViT"],
   "model_type": "efficientnetb0_Vit_blocks_multi_head_attention",
   "framework": "pytorch",
-  "architecture": {
-    "input": {
-      "video_frames":"20 frames per video" ,
-      "image_size": [224, 224]
-      
-    },
-    "output_classes": ["real", "fake"]
-  },
-  "pretrained": true,
-  "model_file": "model.pth"
+  "precision": "float32",
+  "num_parameters": 20026725,
+  "model_format": "safetensors",
+  "license": "mit",
+  "tags": [
+    "image-classification",
+    "video-analysis",
+    "deepfake-detection",
+    "efficientvit",
+    "pytorch"
+  ]
 }