Spaces:
Runtime error
Runtime error
Updating genism syntax due to method deprecation
Browse files
app.py
CHANGED
|
@@ -51,8 +51,8 @@ def syns(word):
|
|
| 51 |
|
| 52 |
|
| 53 |
w2v = dict({})
|
| 54 |
-
for idx, key in enumerate(glove_vectors.
|
| 55 |
-
w2v[key] = glove_vectors.
|
| 56 |
|
| 57 |
|
| 58 |
def calculate_diversity(text):
|
|
@@ -162,7 +162,7 @@ def stats(text):
|
|
| 162 |
|
| 163 |
def predict(text, tokenizer=tokenizer):
|
| 164 |
model.eval()
|
| 165 |
-
model.to(
|
| 166 |
|
| 167 |
def prepare_data(text, tokenizer):
|
| 168 |
input_ids = []
|
|
@@ -188,8 +188,8 @@ def predict(text, tokenizer=tokenizer):
|
|
| 188 |
tokenized_example_text = prepare_data(text, tokenizer)
|
| 189 |
with torch.no_grad():
|
| 190 |
result = model(
|
| 191 |
-
tokenized_example_text['input_ids'].to(
|
| 192 |
-
attention_mask=tokenized_example_text['attention_masks'].to(
|
| 193 |
return_dict=True
|
| 194 |
).logits
|
| 195 |
|
|
|
|
| 51 |
|
| 52 |
|
| 53 |
w2v = dict({})
|
| 54 |
+
for idx, key in enumerate(glove_vectors.key_to_index.keys()):
|
| 55 |
+
w2v[key] = glove_vectors.get_vector(key)
|
| 56 |
|
| 57 |
|
| 58 |
def calculate_diversity(text):
|
|
|
|
| 162 |
|
| 163 |
def predict(text, tokenizer=tokenizer):
|
| 164 |
model.eval()
|
| 165 |
+
model.to('cpu')
|
| 166 |
|
| 167 |
def prepare_data(text, tokenizer):
|
| 168 |
input_ids = []
|
|
|
|
| 188 |
tokenized_example_text = prepare_data(text, tokenizer)
|
| 189 |
with torch.no_grad():
|
| 190 |
result = model(
|
| 191 |
+
tokenized_example_text['input_ids'].to('cpu'),
|
| 192 |
+
attention_mask=tokenized_example_text['attention_masks'].to('cpu'),
|
| 193 |
return_dict=True
|
| 194 |
).logits
|
| 195 |
|