Update README.md

README.md

@@ -70,12 +70,12 @@ The best in MantisScore series is in bold and the best in baselines is underline
 
 ## Usage
 ### Installation
-```bash
+```
 git clone https://github.com/TIGER-AI-Lab/MantisScore.git
 ```
 
 ### Inference
-```shell
+```
 cd MantisScore/examples
 ```
 
@@ -83,6 +83,7 @@ cd MantisScore/examples
 import av
 import numpy as np
 from typing import List
+from PIL import Image
 import torch
 from transformers import AutoProcessor
 from models.idefics2 import Idefics2ForSequenceClassification
@@ -128,11 +129,12 @@ For this video, the text prompt is "{text_prompt}",
 all the frames of video are as follows:
 """
 
+model_name="TIGER-Lab/MantisScore"
 video_path="video1.mp4"
 video_prompt="Near the Elephant Gate village, they approach the haunted house at night. Rajiv feels anxious, but Bhavesh encourages him. As they reach the house, a mysterious sound in the air adds to the suspense."
 
-processor = AutoProcessor.from_pretrained(f"TIGER-Lab/MantisScore",torch_dtype=torch.bfloat16)
-model = Idefics2ForSequenceClassification.from_pretrained(f"TIGER-Lab/MantisScore",torch_dtype=torch.bfloat16).eval()
+processor = AutoProcessor.from_pretrained(model_name,torch_dtype=torch.bfloat16)
+model = Idefics2ForSequenceClassification.from_pretrained(model_name,torch_dtype=torch.bfloat16).eval()
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)