Update README.md

README.md

@@ -5,10 +5,10 @@ language:
 pipeline_tag: image-to-text
 widget:
   - src: >-
-      https://huggingface.co/snzhang/FileTitle-Beit-GPT2/resolve/main/SpiderMan.jpg
+      https://huggingface.co/snzhang/FilmTitle-Beit-GPT2/resolve/main/SpiderMan.jpg
     example_title: SpiderMan
   - src: >-
-      https://huggingface.co/snzhang/FileTitle-Beit-GPT2/resolve/main/BorntoFly.jpg
+      https://huggingface.co/snzhang/FilmTitle-Beit-GPT2/resolve/main/BorntoFly.jpg
     example_title: Born to Fly
 ---
 
@@ -28,7 +28,7 @@ The training data contains 5043 movie posts and their corresponding Chinese titl
 from transformers import VisionEncoderDecoderModel, ViTFeatureExtractor, AutoTokenizer
 from PIL import Image
 
-pretrained = "snzhang/FileTitle-Beit-GPT2"
+pretrained = "snzhang/FilmTitle-Beit-GPT2"
 model = VisionEncoderDecoderModel.from_pretrained(pretrained)
 feature_extractor = ViTFeatureExtractor.from_pretrained(pretrained)
 tokenizer = AutoTokenizer.from_pretrained(pretrained)
@@ -47,4 +47,4 @@ print(preds)
 
 ## More Details
 
-You can get more training details in [FileTitle-Beit-GPT2](https://github.com/h7nian/FileTitle-Beit-GPT2)
+You can get more training details in [FilmTitle-Beit-GPT2](https://github.com/h7nian/FilmTitle-Beit-GPT2)