Update app.py

app.py

@@ -3,6 +3,8 @@ from transformers import AutoProcessor, AutoTokenizer, AutoImageProcessor, AutoM
 import torch
 import open_clip
 
+from huggingface_hub import hf_hub_download
+
 torch.hub.download_url_to_file('http://images.cocodataset.org/val2017/000000039769.jpg', 'cats.jpg')
 torch.hub.download_url_to_file('https://huggingface.co/datasets/nielsr/textcaps-sample/resolve/main/stop_sign.png', 'stop_sign.png')
 torch.hub.download_url_to_file('https://cdn.openai.com/dall-e-2/demos/text2im/astronaut/horse/photo/0.jpg', 'astronaut.jpg')
@@ -23,9 +25,11 @@ vitgpt_processor = AutoImageProcessor.from_pretrained("nlpconnect/vit-gpt2-image
 vitgpt_model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
 vitgpt_tokenizer = AutoTokenizer.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
 
+filepath = hf_hub_download(repo_id="gpucce/CoCa", filename=""laion2B-s13B-b90k-mscoco-2014.pt")
+
 coca_model, _, coca_transform = open_clip.create_model_and_transforms(
     "coca_ViT-L-14",
-    pretrained="laion2B-s13B-b90k-mscoco-2014.pt"
+    pretrained=filepath,
 )
 
 device = "cuda" if torch.cuda.is_available() else "cpu"