Ahsen Khaliq commited on
Commit
38ab5be
β€’
1 Parent(s): 5abcb97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ import gradio as gr
21
  torch.hub.download_url_to_file('https://dl.fbaipublicfiles.com/pytorchvideo/projects/archery.mp4', 'archery.mp4')
22
 
23
  model_name = 'x3d_s'
24
- model = torch.hub.load('facebookresearch/pytorchvideo', model_name, pretrained=True)
25
 
26
  # Set to GPU or CPU
27
  device = "cuda"
21
  torch.hub.download_url_to_file('https://dl.fbaipublicfiles.com/pytorchvideo/projects/archery.mp4', 'archery.mp4')
22
 
23
  model_name = 'x3d_s'
24
+ model = torch.hub.load('facebookresearch/pytorchvideo:main', model_name, pretrained=True)
25
 
26
  # Set to GPU or CPU
27
  device = "cuda"