yuhe6 commited on
Commit
bd4ec9e
1 Parent(s): 250da89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -23,8 +23,8 @@ n_categories = len(model_categories)
23
  # Download an example image from the pytorch website
24
  torch.hub.download_url_to_file("https://upload.wikimedia.org/wikipedia/commons/5/5b/Dog_%28Canis_lupus_familiaris%29_%281%29.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([
 
23
  # Download an example image from the pytorch website
24
  torch.hub.download_url_to_file("https://upload.wikimedia.org/wikipedia/commons/5/5b/Dog_%28Canis_lupus_familiaris%29_%281%29.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://upload.wikimedia.org/wikipedia/commons/4/4d/Cat_March_2010-1.jpg", "cat1.jpg")
27
+ torch.hub.download_url_to_file("https://upload.wikimedia.org/wikipedia/commons/4/4d/Cat_November_2010-1a.jpg", "cat2.jpg")
28
 
29
  def inference(input_image):
30
  preprocess = transforms.Compose([