Update handler.py

handler.py

@@ -31,7 +31,7 @@ class EndpointHandler():
         
         # run inference pipeline
         with autocast(device.type):
-            image = self.pipe(inputs, guidance_scale=7.5)["sample"][0]  
+            image = self.pipe(inputs, guidance_scale=7.5).images[0]
             
         # encode image as base 64
         buffered = BytesIO()