Update app.py

app.py

@@ -3,11 +3,11 @@ from PIL import Image
 from transformers import CLIPModel, AutoTokenizer, AutoProcessor
 import torch
 
-# Load Jina CLIP model
+# Load Jina CLIP model with trust_remote_code=True
 model_name = "jinaai/jina-clip-v1"
-model = CLIPModel.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-processor = AutoProcessor.from_pretrained(model_name)
+model = CLIPModel.from_pretrained(model_name, trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
+processor = AutoProcessor.from_pretrained(model_name, trust_remote_code=True)
 
 def compute_similarity(input1, input2, type1, type2):
     inputs = []