Update handler.py

handler.py

@@ -14,6 +14,10 @@ logger.setLevel(logging.DEBUG)
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 subprocess.run("nvidia-smi")
 
+logger.info(f"torch version: {torch.__version__}")
+logger.info(f"diffusers version: {diffusers.__version__}")
+logger.info(f"transformers version: {transformers.__version__}")
+
 logger.info(f"device: {device}")
 
 class EndpointHandler():
@@ -44,7 +48,6 @@ class EndpointHandler():
 
         parameters = data.pop("parameters", data)
         prompt = parameters['prompt']
-        subprocess.run("nvidia-smi")
 
         images = self.pipe(prompt, image=image, num_inference_steps=10, image_guidance_scale=1).images