rocioadlc commited on
Commit
ec812c4
Β·
verified Β·
1 Parent(s): 32955ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -45,8 +45,9 @@ from tensorflow import keras
45
  from PIL import Image
46
 
47
  # Cell 1: Image Classification Model
48
- model1 = from_pretrained_keras("rocioadlc/model")
49
 
 
50
  class_labels = ['battery',
51
  'biological',
52
  'brown-glass',
 
45
  from PIL import Image
46
 
47
  # Cell 1: Image Classification Model
48
+ model1 = from_pretrained_keras("rocioadlc/EfficientNetV2L")
49
 
50
+ # Define class labels
51
  class_labels = ['battery',
52
  'biological',
53
  'brown-glass',