Spaces:

ashutoshsharma58
/

project1

Build error

ashutoshsharma58 commited on Jan 8, 2024

Commit

9a277df

1 Parent(s): 237ea5f

Update classifier.py

Files changed (1) hide show

classifier.py CHANGED Viewed

@@ -15,7 +15,7 @@ from tensorflow.keras.applications.efficientnet import preprocess_input
 # Load trained model
 model = load_model('cnn_model.h5', compile=False)
 # Define class labels
-class_labels = ['Gasoline Can','Hammer', 'Pebbles','Pliers', 'Rope', 'Screwdriver', 'Toolbox', 'Wrench/Spanner']
 def predict_and_display(uploaded_file, model, class_labels):
     img = Image.open(uploaded_file)

 # Load trained model
 model = load_model('cnn_model.h5', compile=False)
 # Define class labels
+class_labels = ['Gasoline Can','Hammer', 'Pebbles','Pliers', 'Rope', 'Screwdriver', 'Toolbox', 'Wrench or Spanner']
 def predict_and_display(uploaded_file, model, class_labels):
     img = Image.open(uploaded_file)