Update app.py

app.py

@@ -3,11 +3,11 @@ 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
+#https://huggingface.co/spaces/yuhe6/final_project/blob/main/Net_Rotate9.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)
-torch.hub.download_url_to_file('https://huggingface.co/spaces/yuhe6/final_project/blob/main/CIFAR10_cnn.pth', '/tmp/temporary_file')
+torch.hub.download_url_to_file('https://huggingface.co/spaces/yuhe6/final_project/blob/main/Net_Rotate9.pth', '/tmp/temporary_file')
 model = torch.hub.load('/tmp', 'temporary_file', pretrained=True)
 
 model.eval()