Update app.py

app.py

@@ -40,9 +40,18 @@ spaBERT_model.to(device)
 spaBERT_model.eval()
 
 
-
-
-
+#Get BERT Embedding for review
+def get_bert_embedding(review_text)
+    #tokenize review
+    inputs = bert_tokenizer(review_text, return_tensors='pt', padding=True, truncation=True).to(device)
+    
+    # Forward pass through the BERT model
+    with torch.no_grad():
+        outputs = bert_model(**inputs)
+
+    # Extract embeddings from the last hidden state
+    embeddings = outputs.last_hidden_state[:, 0, :].detach()     #CLS Token
+    return embeddings