Update app.py

app.py

@@ -55,7 +55,7 @@ def load_model(file_name: str, device: torch.device):
     return model
     
 def generate_image(model_name: str, seed: int, truncation_psi: float):
-   device = 'cuda'
+   device = 'cpu'
    model = model_dict[model_name]
    model = load_model(model, device)
    seed = int(np.clip(seed, 0, np.iinfo(np.uint32).max))