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 |
|