Spaces:

Tasty-Rice
/

Magic_on_paper

Runtime error

TastyRice commited on May 9, 2024

Commit

0caf86f

verified ·

1 Parent(s): 5bff417

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,7 +36,8 @@ MODEL = os.getenv(
 torch.backends.cudnn.deterministic = True
 torch.backends.cudnn.benchmark = False
-device = torch.device("cuda:0" if torch.cuda.is_available() else torch.cpu.device.is_available())
 def load_pipeline(model_name):

 torch.backends.cudnn.deterministic = True
 torch.backends.cudnn.benchmark = False
+device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
+    model = model.to(device)
 def load_pipeline(model_name):