Update XGBoost_utils.py

XGBoost_utils.py

@@ -595,7 +595,6 @@ def RMSRPD(y_pred, y_true):
 
 def Caption_Generation(image): #image is a PIL Image object
     with torch.no_grad():
-        image = image.convert('RGB')
         model = AutoModel.from_pretrained('openbmb/MiniCPM-V-2', trust_remote_code=True, torch_dtype=torch.float16)
         model = model.to(device='cpu', dtype=torch.float32).eval()