Changed a better model trained

app.py

@@ -147,7 +147,7 @@ class BigramLanguageModel(nn.Module):
         return idx
 
 model = BigramLanguageModel()
-model.load_state_dict(torch.load('trainedstorycharmodel.pt', map_location=torch.device('cpu')))
+model.load_state_dict(torch.load('StoryCharModelOptimized.pt', map_location=torch.device('cpu')))
 # m = model.to(device)
 # print the number of parameters in the model
 # print(sum(p.numel() for p in m.parameters())/1e6, 'M parameters')