Hugging Face's logo

Spaces:
serdaryildiz
/
MViT-TR
Sleeping

App Files Community
serdaryildiz commited on
Commit
0f4cd8f
·
verified ·
1 Parent(s): a33b529

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ model = TTR({"img_size": [32, 128],
28
  })
29
  model.load_state_dict(torch.load(model_path, map_location=device)["model"])
30
  model.eval()
31
- model = model.to("cuda:0")
32
 
33
  preprocess = getTransforms()
34
 
 
28
  })
29
  model.load_state_dict(torch.load(model_path, map_location=device)["model"])
30
  model.eval()
31
+ model = model.to(device)
32
 
33
  preprocess = getTransforms()
34