nullHawk commited on
Commit
d83ec05
·
verified ·
1 Parent(s): 41ce832

fixed word embedding issue

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def run(name):
12
  hidden_tensor = rnn.init_hidden()
13
  for i in range(input_tensor.size()[0]):
14
  output, hidden_tensor = rnn(input_tensor[i], hidden_tensor)
15
- return f"It is an {language_map[torch.argmax(output).item()]} name"
16
 
17
  rnn = model.RNN(N_LETTERS, 128, 19)
18
  rnn.load_state_dict(torch.load('rnn.pth'))
 
12
  hidden_tensor = rnn.init_hidden()
13
  for i in range(input_tensor.size()[0]):
14
  output, hidden_tensor = rnn(input_tensor[i], hidden_tensor)
15
+ return f"It is {language_map[torch.argmax(output).item()]} name"
16
 
17
  rnn = model.RNN(N_LETTERS, 128, 19)
18
  rnn.load_state_dict(torch.load('rnn.pth'))