Update README.md (#2)

- Update README.md (e4a4d74f37c3c6ddda5717384463dcc713ce528d)


Co-authored-by: Bertug Mirasyedi <bertugmirasyedi@users.noreply.huggingface.co>

README.md

@@ -72,7 +72,11 @@ from model import CustomBertModel
 tokenizer = BertTokenizer.from_pretrained('Berketarak/Product-Matching-Classifier')
 model = CustomBertModel.from_pretrained('Berketarak/Product-Matching-Classifier')
 
+# Send model to GPU
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+model.to(device)
 
+# Define the products
 product1 = 'X brand Pegasus Sneakers'
 product2 = 'Y brand Shoes'
 
@@ -91,4 +95,3 @@ if output > 0.5:
     print(f"The products are likely the SAME. Model output: {output}")
 else:
     print(f"The products are likely DIFFERENT. Model output: {output}")
-