Update app.py

app.py

@@ -20,7 +20,7 @@ encode = lambda s: [stoi[c] for c in s] # encoder: take a string, output a list
 decode = lambda l: ''.join([itos[i] for i in l]) # decoder: take a list of integers, output a string
 
 # load model
-model_spec = torch.load("ckpt.pt")
+model_spec = torch.load("ckpt.pt", map_location=torch.device('cpu'))
 model_args = model_spec['model_args']
 model_weights = model_spec['model']
 modelconf = ModelConfig(**model_args)