RobPruzan commited on
Commit
2edd6fc
1 Parent(s): 4b9a98d

Updating genism syntax due to method deprecation

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -51,8 +51,8 @@ def syns(word):
51
 
52
 
53
  w2v = dict({})
54
- for idx, key in enumerate(glove_vectors.wv.vocab):
55
- w2v[key] = glove_vectors.wv.get_vector(key)
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(device)
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(device),
192
- attention_mask=tokenized_example_text['attention_masks'].to(device),
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