yuhe6 commited on
Commit
9c2ac28
·
1 Parent(s): 0a84bc5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import gradio as gr
7
  os.system("wget https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt")
8
 
9
  #model = torch.hub.load('huawei-noah/ghostnet', 'ghostnet_1x', pretrained=True)
10
- model = torch.hub.load('', 'CIFAR10_cnn', pretrained=True)
11
 
12
  model.eval()
13
  # Download an example image from the pytorch website
 
7
  os.system("wget https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt")
8
 
9
  #model = torch.hub.load('huawei-noah/ghostnet', 'ghostnet_1x', pretrained=True)
10
+ model = torch.hub.load('/', 'CIFAR10_cnn', pretrained=True)
11
 
12
  model.eval()
13
  # Download an example image from the pytorch website