ma4389 commited on
Commit
1e3bcc3
·
verified ·
1 Parent(s): 023f391

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ val_transforms = transforms.Compose([
31
  ])
32
 
33
  # 🏷️ Class names
34
- class_names = ["dog", "wild", "cat"]
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):