Update app.py

app.py

@@ -5,7 +5,7 @@ from mingru_lm import MinGRU_LM
 
 model = MinGRU_LM(dim=512, num_tokens=256, num_layers=6)
 pt_model = "best_model.pt"
-checkpoint = torch.load(pt_model)
+checkpoint = torch.load(pt_model,map_location=torch.device('cpu'))
 model.load_state_dict(checkpoint['model_state_dict'])
 
 # Move model to GPU if available