Update README.md

README.md

@@ -113,6 +113,11 @@ torch.set_float32_matmul_precision(['high', 'highest'][0])
 birefnet.to('cuda')
 birefnet.eval()
 
+print('BiRefNet is ready to use.')
+
+birefnet.half()
+print('Using half precision!')
+
 def extract_object(birefnet, imagepath):
     # Data settings
     image_size = (1024, 1024)
@@ -124,6 +129,7 @@ def extract_object(birefnet, imagepath):
 
     image = Image.open(imagepath)
     input_images = transform_image(image).unsqueeze(0).to('cuda')
+    input_images = input_images.half()
 
     # Prediction
     with torch.no_grad():