Update handler.py

handler.py

@@ -45,8 +45,8 @@ class EndpointHandler():
         logger.info('image_bytes')
         image = Image.open(image_bytes)
         logger.info('image')
-        #with autocast(device.type):
-        #    upscaled_image = self.pipe(prompt="", image = decoded_image).images[0]
+        with autocast(device.type):
+            upscaled_image = self.pipe(prompt="", image = image).images[0]
 
         #buffered = BytesIO()
         #upscaled_image.save(buffered, format="JPEG")