update model device

app.py

@@ -145,7 +145,7 @@ decode = lambda l: ''.join([itos[i] for i in l]) # decoder: take a list of integ
 # ----------------------------------------------------------------
 
 # load model
-model = torch.load('complete-model.pt')
+model = torch.load('complete-model.pt', map_location=device)
 
 # inference
 slider_value = st.slider('Amount of text to generate', min_value=100, max_value=2000, value=500, step=5)