Update app.py

app.py

@@ -18,14 +18,15 @@ with open('./intents.json', 'r') as json_data:
     intents = json.load(json_data)
 
 FILE = "./data.pth"
-data = torch.load(FILE, map_location=torch.device('cpu'))
+data = torch.load(FILE)
+model_state = torch.load(FILE['model_state'], map_location=torch.device('cpu'))
 
 input_size = data["input_size"]
 hidden_size = data["hidden_size"]
 output_size = data["output_size"]
 all_words = data['all_words']
 tags = data['tags']
-model_state = data["model_state"]
+#model_state = data["model_state"]
 
 model = NeuralNet(input_size, hidden_size, output_size).to(device)
 model.load_state_dict(model_state)