Hugging Face's logo

Spaces:
rocioadlc
/
Green_Greta
Sleeping

App Files Community
rocioadlc commited on
Commit
6a01c1a
•
1 Parent(s): d14ae0d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ from tensorflow import keras
47
  from PIL import Image
48
 
49
  # Cell 1: Image Classification Model
50
- model1 = from_pretrained_keras("rocioadlc/efficientnet_trash")
51
 
52
  # Define class labels
53
  class_labels = ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']
 
47
  from PIL import Image
48
 
49
  # Cell 1: Image Classification Model
50
+ model1 = from_pretrained_keras("rocioadlc/efficientnetB0_trash")
51
 
52
  # Define class labels
53
  class_labels = ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']