Update app.py

app.py

@@ -3,10 +3,12 @@ import torch
 from PIL import Image
 from torchvision import transforms
 import gradio as gr
-
+#https://huggingface.co/spaces/yuhe6/final_project/blob/main/CIFAR10_cnn.pth
 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('huawei-noah/ghostnet', 'ghostnet_1x', pretrained=True)
+model = torch.hub.load('', 'CIFAR10_cnn', pretrained=True)
+
 model.eval()
 # Download an example image from the pytorch website
 torch.hub.download_url_to_file("https://github.com/pytorch/hub/raw/master/images/dog.jpg", "dog.jpg")