DongfuJiang commited on
Commit
e02bc9d
1 Parent(s): 06cb631
Files changed (1) hide show
  1. app_regression.py +2 -2
app_regression.py CHANGED
@@ -14,8 +14,8 @@ from models.conversation import conv_templates
14
  from typing import List
15
 
16
 
17
- processor = AutoProcessor.from_pretrained("TIGER-Lab/MantisScore")
18
- model = Idefics2ForSequenceClassification.from_pretrained("TIGER-Lab/MantisScore", torch_dtype=torch.bfloat16).eval()
19
 
20
  MAX_NUM_FRAMES = 24
21
  conv_template = conv_templates["idefics_2"]
 
14
  from typing import List
15
 
16
 
17
+ processor = AutoProcessor.from_pretrained("TIGER-Lab/VideoScore")
18
+ model = Idefics2ForSequenceClassification.from_pretrained("TIGER-Lab/VideoScore", torch_dtype=torch.bfloat16).eval()
19
 
20
  MAX_NUM_FRAMES = 24
21
  conv_template = conv_templates["idefics_2"]