fixed word embedding issue

app.py

@@ -3,9 +3,7 @@ import torch
 import model
 from utils import line_to_tensor, load_data, N_LETTERS
 
-def category_from_output(output):
-    category_index = torch.argmax(output).item()
-    return all_categories[category_index]
+language_map = ['Polish', 'Irish', 'Italian', 'Korean', 'English', 'Czech', 'Chinese', 'Japanese', 'Portuguese', 'Indian', 'Greek', 'Vietnamese', 'French', 'German', 'Russian', 'Scottish', 'Arabic', 'Dutch', 'Spanish']
 
 def run(name):
     
@@ -14,10 +12,9 @@ def run(name):
         hidden_tensor = rnn.init_hidden()
         for i in range(input_tensor.size()[0]):
             output, hidden_tensor = rnn(input_tensor[i], hidden_tensor)
-        return f"It is an {category_from_output(output)} name"
+        return f"It is an {language_map[torch.argmax(output).item()]} name"
 
-category_lines, all_categories = load_data()
-rnn = model.RNN(N_LETTERS, 128, len(all_categories))
+rnn = model.RNN(N_LETTERS, 128, 19)
 rnn.load_state_dict(torch.load('rnn.pth'))
 rnn.eval