Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -31,7 +31,7 @@ val_transforms = transforms.Compose([
|
|
| 31 |
])
|
| 32 |
|
| 33 |
# 🏷️ Class names
|
| 34 |
-
class_names = ["dog", "
|
| 35 |
|
| 36 |
# 🔮 Prediction function
|
| 37 |
def classify_image(img):
|
|
|
|
| 31 |
])
|
| 32 |
|
| 33 |
# 🏷️ Class names
|
| 34 |
+
class_names = ["dog", "cat", "wild"]
|
| 35 |
|
| 36 |
# 🔮 Prediction function
|
| 37 |
def classify_image(img):
|