Added more loggers

handler.py

@@ -20,11 +20,13 @@ class EndpointHandler():
         # self.model = self.model.to(device)
 
 
-
+        print(" $$$$ Model Loading $$$$")
         self.processor = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b")
         self.model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b")
+        print(" $$$$ model loaded $$$$")
         self.model.eval()
         self.model = self.model.to(device)
+        print(" $$$$ model loaded 2222 $$$$")
         
         
 
@@ -40,8 +42,8 @@ class EndpointHandler():
         """
         print("********* Helllo ***********")
         print(data)
-        img_data = data.pop("input", data)
-        prompt = data.pop("prompt", None)
+        img_data = data.pop("inputs", data)
+        prompt = data.pop("prompt", "")
         print("#########")
         # parameters = data.pop("parameters", {})