Spaces:
Sleeping
Sleeping
fixed word embedding issue
Browse files
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
|
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'))
|