yuhe6 commited on
Commit
bb423ad
·
1 Parent(s): 950a72b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -21,10 +21,10 @@ n_categories = len(model_categories)
21
 
22
  #model.eval()
23
  # Download an example image from the pytorch website
24
- torch.hub.download_url_to_file("https://huggingface.co/spaces/yuhe6/final_project/blob/main/american_bulldog_5.jpg", "dog1.jpg")
25
  torch.hub.download_url_to_file("https://upload.wikimedia.org/wikipedia/commons/thumb/6/6e/Golde33443.jpg/640px-Golde33443.jpg", "dog2.jpg")
26
- torch.hub.download_url_to_file("https://huggingface.co/spaces/yuhe6/final_project/blob/main/Bombay_5.jpg", "cat1.jpg")
27
- torch.hub.download_url_to_file("https://huggingface.co/spaces/yuhe6/final_project/blob/main/Birman_5.jpg", "cat2.jpg")
28
 
29
  def inference(input_image):
30
  preprocess = transforms.Compose([
 
21
 
22
  #model.eval()
23
  # Download an example image from the pytorch website
24
+ torch.hub.download_url_to_file("https://github.com/yuhe6/Stat-430-Deep-Learning/blob/main/american_bulldog_5.jpg", "dog1.jpg")
25
  torch.hub.download_url_to_file("https://upload.wikimedia.org/wikipedia/commons/thumb/6/6e/Golde33443.jpg/640px-Golde33443.jpg", "dog2.jpg")
26
+ torch.hub.download_url_to_file("https://github.com/yuhe6/Stat-430-Deep-Learning/blob/main/Bombay_5.jpg", "cat1.jpg")
27
+ torch.hub.download_url_to_file("https://github.com/yuhe6/Stat-430-Deep-Learning/blob/main/Birman_5.jpg", "cat2.jpg")
28
 
29
  def inference(input_image):
30
  preprocess = transforms.Compose([