adding logs

main.py

@@ -124,6 +124,8 @@ def predict(data: PredictRequest):
         dict: The generated description of the image(s).
     """
 
+    print("Calling /predict endpoint...")
+
     # Ensure image_base64 is a list (even if a single image is provided)
     image_list = (
         data.image_base64
@@ -156,6 +158,8 @@ def predict(data: PredictRequest):
         return_tensors="pt",
     ).to(model.device)
 
+    print("Starting generation...")
+
     # Generate the output
     generated_ids = model.generate(**inputs, max_new_tokens=2056)
     generated_ids_trimmed = [