Hugging Face's logo

Spaces:
paloma99
/
test-space-2
Sleeping

App Files Community
paloma99 commited on
Commit
1ed0e03
·
verified ·
1 Parent(s): 0b27d99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -1
app.py CHANGED
@@ -50,7 +50,18 @@ from PIL import Image
50
  model1 = from_pretrained_keras("rocioadlc/EfficientNetV2L")
51
 
52
  # Define class labels
53
- class_labels = ['cardboard', 'compost', 'glass', 'metal', 'paper', 'plastic', 'trash']
 
 
 
 
 
 
 
 
 
 
 
54
 
55
  # Function to predict image label and score
56
  def predict_image(input):
 
50
  model1 = from_pretrained_keras("rocioadlc/EfficientNetV2L")
51
 
52
  # Define class labels
53
+ class_labels = ['battery',
54
+ 'biological',
55
+ 'brown-glass',
56
+ 'cardboard',
57
+ 'clothes',
58
+ 'green-glass',
59
+ 'metal',
60
+ 'paper',
61
+ 'plastic',
62
+ 'shoes',
63
+ 'trash',
64
+ 'white-glass']
65
 
66
  # Function to predict image label and score
67
  def predict_image(input):