Updated model cards

README.md

@@ -108,13 +108,13 @@ from huggingface_hub import hf_hub_download
 
 # Recommended: final optimized models
 ckpt_path = hf_hub_download(
-    repo_id="nauthui7/school-violence-detection-models",
+    repo_id="visionlab-ai/school-violence-detection-models",
     filename="final/final_x3d_realtime.pt",   # or "final/final_best.pt"
 )
 
 # Or one of the baselines
 ckpt_path = hf_hub_download(
-    repo_id="nauthui7/school-violence-detection-models",
+    repo_id="visionlab-ai/school-violence-detection-models",
     filename="x3d_m/x3d_m_best.pt",
 )
 ```
@@ -194,6 +194,6 @@ MIT License - tự do sử dụng cho mục đích nghiên cứu và phi thươn
 
 ## Contact
 
-- Author: nauthui7
-- Demo Space: [nauthui7/school-violence-detection-app](https://huggingface.co/spaces/nauthui7/school-violence-detection-app)
+- Author: visionlab-ai
+- Demo Space: [visionlab-ai/school-violence-detection-app](https://huggingface.co/spaces/visionlab-ai/school-violence-detection-app)
 - Source code: GitHub repository (ask author for link)

final/MODEL_CARD.md

@@ -82,7 +82,7 @@ from sv_model import create_model
 
 # X3D-M realtime
 ckpt_path = hf_hub_download(
-    repo_id="nauthui7/school-violence-detection-models",
+    repo_id="visionlab-ai/school-violence-detection-models",
     filename="final/final_x3d_realtime.pt",
 )
 model = create_model(backbone="x3d_m", num_classes=2, pretrained=False, dropout=0.3)
@@ -92,7 +92,7 @@ model.eval()
 
 # R(2+1)D-18 best
 ckpt_path = hf_hub_download(
-    repo_id="nauthui7/school-violence-detection-models",
+    repo_id="visionlab-ai/school-violence-detection-models",
     filename="final/final_best.pt",
 )
 model = create_model(backbone="r2plus1d_18", num_classes=2, pretrained=False, dropout=0.5)

i3d_r50/MODEL_CARD.md

@@ -35,7 +35,7 @@ from huggingface_hub import hf_hub_download
 from pytorchvideo.models.hub import i3d_r50
 
 ckpt_path = hf_hub_download(
-    repo_id="nauthui7/school-violence-detection-models",
+    repo_id="visionlab-ai/school-violence-detection-models",
     filename="i3d_r50/i3d_r50_best.pt",
 )
 

r2plus1d_18/MODEL_CARD.md

@@ -35,7 +35,7 @@ from huggingface_hub import hf_hub_download
 from torchvision.models.video import r2plus1d_18
 
 ckpt_path = hf_hub_download(
-    repo_id="nauthui7/school-violence-detection-models",
+    repo_id="visionlab-ai/school-violence-detection-models",
     filename="r2plus1d_18/r2plus1d_best.pt",
 )
 

slowfast_r50/MODEL_CARD.md

@@ -37,7 +37,7 @@ import torch
 from huggingface_hub import hf_hub_download
 
 ckpt_path = hf_hub_download(
-    repo_id="nauthui7/school-violence-detection-models",
+    repo_id="visionlab-ai/school-violence-detection-models",
     filename="slowfast_r50/slowfast_best.pt",
 )
 

x3d_m/MODEL_CARD.md

@@ -35,7 +35,7 @@ from huggingface_hub import hf_hub_download
 from pytorchvideo.models.hub import x3d_m
 
 ckpt_path = hf_hub_download(
-    repo_id="nauthui7/school-violence-detection-models",
+    repo_id="visionlab-ai/school-violence-detection-models",
     filename="x3d_m/x3d_m_best.pt",
 )