Spaces:
Runtime error
Runtime error
Commit
•
fb00e8c
1
Parent(s):
d0b6dd3
update
Browse files- README.md +2 -2
- app_regression.py +2 -2
- logo3.png +0 -0
README.md
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
---
|
2 |
-
title:
|
3 |
-
emoji:
|
4 |
colorFrom: green
|
5 |
colorTo: yellow
|
6 |
sdk: gradio
|
|
|
1 |
---
|
2 |
+
title: MantisScore
|
3 |
+
emoji: ![alt text](logo3.png)
|
4 |
colorFrom: green
|
5 |
colorTo: yellow
|
6 |
sdk: gradio
|
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("
|
18 |
-
model = Idefics2ForSequenceClassification.from_pretrained("
|
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/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"]
|
logo3.png
ADDED
![]() |