update

README.md

@@ -1,6 +1,6 @@
 ---
-title: Video-Eval-Bench
-emoji: 📹
+title: MantisScore
+emoji: ![alt text](logo3.png)
 colorFrom: green
 colorTo: yellow
 sdk: gradio

app_regression.py

@@ -14,8 +14,8 @@ from models.conversation import conv_templates
 from typing import List
 
 
-processor = AutoProcessor.from_pretrained("Mantis-VL/mantis-8b-idefics2-video-eval-refined-40k-ablation-anno_4096_regression")
-model = Idefics2ForSequenceClassification.from_pretrained("Mantis-VL/mantis-8b-idefics2-video-eval-refined-40k-ablation-anno_4096_regression", torch_dtype=torch.bfloat16).eval()
+processor = AutoProcessor.from_pretrained("TIGER-Lab/MantisScore")
+model = Idefics2ForSequenceClassification.from_pretrained("TIGER-Lab/MantisScore", torch_dtype=torch.bfloat16).eval()
 
 MAX_NUM_FRAMES = 24
 conv_template = conv_templates["idefics_2"]