Update app.py

app.py

@@ -7,10 +7,11 @@ 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)
-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 = torch.jit.load('https://huggingface.co/spaces/yuhe6/final_project/blob/main/Net_Rotate9.pth').eval().to(device)
+#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()
+#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")