Update app.py

app.py

@@ -60,7 +60,7 @@ def encode_image(image_path):
 
 #Get the projection model
 img_proj_head = _MLPVectorProjector(512, 2560, 1, 4).to("cpu")
-img_proj_head.load_state_dict(torch.load('projection_finetuned.pth'))
+img_proj_head.load_state_dict(torch.load('projection_finetuned.pth',  map_location=torch.device('cpu')))
 
 #Get the fine-tuned phi-2 model
 phi2_finetuned = AutoModelForCausalLM.from_pretrained(