Hugging Face's logo

Spaces:
Tasty-Rice
/
Magic_on_paper
Runtime error

App Files Community
TastyRice commited on
Commit
d9ef08c
•
1 Parent(s): 683c211

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ torch.backends.cudnn.deterministic = True
37
  torch.backends.cudnn.benchmark = False
38
 
39
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
40
- model = model.to(device)
41
 
42
 
43
  def load_pipeline(model_name):
 
37
  torch.backends.cudnn.benchmark = False
38
 
39
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
40
+ model.to(device)
41
 
42
 
43
  def load_pipeline(model_name):