Update handler.py

handler.py

@@ -9,8 +9,6 @@ LOGGER = logging.getLogger(__name__)
 
 class EndpointHandler():
     def __init__(self, path=""):
-        print("CALLED")
-        LOGGER.info(f"Start generation.")
         self.model = AutoModelForCausalLM.from_pretrained("Ozgur98/pushed_model_mosaic_small")
         self.tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neox-20b")
         # Load the Lora model