Update app.py

app.py

@@ -15,7 +15,7 @@ def generate(movie, celebrity, scale):
 HF_TOKEN = os.getenv('HF_TOKEN')
 
 pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=HF_TOKEN)
-device = torch.device('coda' if torch.cuda.is_available() else 'cpu')
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 generator = torch.Generator(device=device)
 pipe = pipe.to(device)