Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,10 +24,6 @@ bert_model.eval()
|
|
| 24 |
data_file_path = 'models/spabert/datasets/SPABERT_finetuning_data_combined.json'
|
| 25 |
pretrained_model_path = 'models/spabert/datasets/fine-spabert-base-uncased-finetuned-osm-mn.pth'
|
| 26 |
|
| 27 |
-
|
| 28 |
-
#b_model = BertForMaskedLM.from_pretrained('bert-base-uncased')
|
| 29 |
-
#b_tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
|
| 30 |
-
|
| 31 |
config = SpatialBertConfig()
|
| 32 |
config.output_hidden_states = True
|
| 33 |
spaBERT_model = SpatialBertForMaskedLM(config)
|
|
@@ -41,7 +37,7 @@ spaBERT_model.eval()
|
|
| 41 |
|
| 42 |
|
| 43 |
#Get BERT Embedding for review
|
| 44 |
-
def get_bert_embedding(review_text)
|
| 45 |
#tokenize review
|
| 46 |
inputs = bert_tokenizer(review_text, return_tensors='pt', padding=True, truncation=True).to(device)
|
| 47 |
|
|
|
|
| 24 |
data_file_path = 'models/spabert/datasets/SPABERT_finetuning_data_combined.json'
|
| 25 |
pretrained_model_path = 'models/spabert/datasets/fine-spabert-base-uncased-finetuned-osm-mn.pth'
|
| 26 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
config = SpatialBertConfig()
|
| 28 |
config.output_hidden_states = True
|
| 29 |
spaBERT_model = SpatialBertForMaskedLM(config)
|
|
|
|
| 37 |
|
| 38 |
|
| 39 |
#Get BERT Embedding for review
|
| 40 |
+
def get_bert_embedding(review_text):
|
| 41 |
#tokenize review
|
| 42 |
inputs = bert_tokenizer(review_text, return_tensors='pt', padding=True, truncation=True).to(device)
|
| 43 |
|