Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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/
|
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([
|