Update app.py

app.py

@@ -7,7 +7,8 @@ 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_state_dict_from_url('https://huggingface.co/spaces/yuhe6/final_project/blob/main/CIFAR10_cnn.pth')
+torch.hub.download_url_to_file('https://huggingface.co/spaces/yuhe6/final_project/blob/main/CIFAR10_cnn.pth', '/tmp/temporary_file')
+model = torch.hub.load('/tmp/', 'temporary_file', pretrained=True)
 
 model.eval()
 # Download an example image from the pytorch website