update

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_4096_regression")
-model = Idefics2ForSequenceClassification.from_pretrained("Mantis-VL/mantis-8b-idefics2-video-eval-refined-40k_4096_regression", torch_dtype=torch.bfloat16).eval()
+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()
 
 MAX_NUM_FRAMES = 24
 conv_template = conv_templates["idefics_2"]