Spaces:

yuhe6
/

final_project

Runtime error

yuhe6 commited on May 2, 2022

Commit

106cd65

1 Parent(s): 4dccbcf

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import gradio as gr
 #os.system("wget https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt")
 #model = torch.hub.load('huawei-noah/ghostnet', 'ghostnet_1x', pretrained=True)
-model = torch.hub.load('/', 'CIFAR10_cnn.pth', pretrained=True)
 model.eval()
 # Download an example image from the pytorch website

 #os.system("wget https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt")
 #model = torch.hub.load('huawei-noah/ghostnet', 'ghostnet_1x', pretrained=True)
+model = torch.hub.load_state_dict_from_url('https://huggingface.co/spaces/yuhe6/final_project/blob/main/CIFAR10_cnn.pth')
 model.eval()
 # Download an example image from the pytorch website