yuhe6 commited on
Commit
d7dcddc
1 Parent(s): bd4ec9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ n_categories = len(model_categories)
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([
 
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/9/9e/Domestic_cat.jpg", "cat2.jpg")
28
 
29
  def inference(input_image):
30
  preprocess = transforms.Compose([