Update app.py

app.py

@@ -41,6 +41,8 @@ model_id_clip = "openai/clip-vit-base-patch16"
 model_clip = CLIPModel.from_pretrained(model_id_clip).to("cpu")
 processor_clip = CLIPProcessor.from_pretrained(model_id_clip)
 
+print('--------------Loaded CLIP----------------------')
+
 # Preprocess the image for clip
 def preprocess_image(image_path):
     image = Image.open(image_path).convert("RGB")
@@ -62,10 +64,14 @@ def encode_image(image_path):
 img_proj_head = _MLPVectorProjector(512, 2560, 1, 4).to("cpu")
 img_proj_head.load_state_dict(torch.load('projection_finetuned.pth',  map_location=torch.device('cpu')))
 
+print('--------------Loaded proj head----------------------')
+
 #Get the fine-tuned phi-2 model
 phi2_finetuned = AutoModelForCausalLM.from_pretrained(
     "phi2_adaptor_fineTuned", trust_remote_code=True,
     torch_dtype = torch.float32).to("cpu")
+
+print('--------------Loaded fine tuned phi2 model----------------------')
     
 
 def example_inference(input_text, count): #, image, img_qn, audio):