Update app.py

app.py

@@ -2,7 +2,7 @@ import torch
 from PIL import Image
 import gradio as gr
 import spaces
-from transformers import AutoProcessor, AutoModel
+from transformers import AutoProcessor, AutoModelForImageTextRetrieval
 import torch.nn.functional as F
 
 #---------------------------------
@@ -13,7 +13,7 @@ def load_biomedclip_model():
     """Loads the BiomedCLIP model and tokenizer."""
     biomedclip_model_name = 'microsoft/BiomedCLIP-PubMedBERT_256-vit_base_patch16_224'
     processor = AutoProcessor.from_pretrained(biomedclip_model_name)
-    model = AutoModel.from_pretrained(biomedclip_model_name).cuda().eval()
+    model = AutoModelForImageTextRetrieval.from_pretrained(biomedclip_model_name).cuda().eval()
     return model, processor
 
 def compute_similarity(image, text, biomedclip_model, biomedclip_processor):