Update README.md

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}")
-