Update app.py

app.py

@@ -21,10 +21,10 @@ n_categories = len(model_categories)
 
 #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", "dog1.jpg")
+torch.hub.download_url_to_file("https://huggingface.co/spaces/yuhe6/final_project/blob/main/american_bulldog_5.jpg", "dog1.jpg")
 torch.hub.download_url_to_file("https://upload.wikimedia.org/wikipedia/commons/thumb/6/6e/Golde33443.jpg/640px-Golde33443.jpg", "dog2.jpg")
-torch.hub.download_url_to_file("https://upload.wikimedia.org/wikipedia/commons/4/4d/Cat_November_2010-1a.jpg", "cat1.jpg")
-torch.hub.download_url_to_file("https://upload.wikimedia.org/wikipedia/en/thumb/7/70/Norwegian_Forest_Cat_in_snow_%28closeup%29_%28cropped%29.jpg/800px-Norwegian_Forest_Cat_in_snow_%28closeup%29_%28cropped%29.jpg", "cat2.jpg")
+torch.hub.download_url_to_file("https://huggingface.co/spaces/yuhe6/final_project/blob/main/Bombay_5.jpg", "cat1.jpg")
+torch.hub.download_url_to_file("https://huggingface.co/spaces/yuhe6/final_project/blob/main/Birman_5.jpg", "cat2.jpg")
 
 def inference(input_image):
     preprocess = transforms.Compose([