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