update

app.py

@@ -10,8 +10,8 @@ import functools
 from transformers import AutoProcessor, Idefics2ForConditionalGeneration
 from models.conversation import conv_templates
 from typing import List
-processor = AutoProcessor.from_pretrained("Mantis-VL/mantis-8b-idefics2-video-eval-95k-mantis-2epoch_4096")
-model = Idefics2ForConditionalGeneration.from_pretrained("Mantis-VL/mantis-8b-idefics2-video-eval-95k-mantis-2epoch_4096")
+processor = AutoProcessor.from_pretrained("Mantis-VL/mantis-8b-idefics2-video-eval-high-res-35k-mantis-2epoch_4096")
+model = Idefics2ForConditionalGeneration.from_pretrained("Mantis-VL/mantis-8b-idefics2-video-eval-high-res-35k-mantis-2epoch_4096", device_map="auto")
 MAX_NUM_FRAMES = 24
 conv_template = conv_templates["idefics_2"]
 
@@ -224,4 +224,4 @@ def build_demo():
 
 if __name__ == "__main__":
     demo = build_demo()
-    demo.launch()
+    demo.launch(share=True)