Hugging Face's logo

Spaces:
pytorch
/
IBN-Net
Runtime error

App Files Community
akhaliq HF Staff commited on
Commit
3c426af
·
1 Parent(s): db4b14b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from torchvision import transforms
4
  import gradio as gr
5
  import os
6
 
7
- model = torch.hub.load('XingangPan/IBN-Net', 'resnet50_ibn_a', pretrained=True)
8
  model.eval()
9
 
10
  # Download ImageNet labels
 
4
  import gradio as gr
5
  import os
6
 
7
+ model = torch.hub.load('AK391/IBN-Net', 'resnet50_ibn_a', pretrained=True)
8
  model.eval()
9
 
10
  # Download ImageNet labels