Spaces:

hshetty
/

movie-poster-generator

Runtime error

hshetty commited on Oct 18, 2022

Commit

c560c8f

1 Parent(s): e518a26

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import torch
 import os
 auth_token = os.environ.get("auth_token")
-device = "cuda"
 generator = torch.Generator(device=device)
 seed = 496012807434005 #generator.seed()

 import os
 auth_token = os.environ.get("auth_token")
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 generator = torch.Generator(device=device)
 seed = 496012807434005 #generator.seed()