Hugging Face's logo

Spaces:
Rishav-ctrl
/
waste-classifier
Runtime error

App Files Community
1
Rishav-ctrl commited on
Commit
299d524
·
verified ·
1 Parent(s): cc41790

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,7 +4,8 @@ from tensorflow.keras.preprocessing import image
4
  import numpy as np
5
 
6
  # Use TFSMLayer to load the SavedModel
7
- model = tf.keras.layers.experimental.preprocessing.TFSMLayer("waste_sort_model", call_endpoint="serving_default")
 
8
  class_names = ["Non-Recyclable", "Recyclable"]
9
 
10
  def classify_image(img):
 
4
  import numpy as np
5
 
6
  # Use TFSMLayer to load the SavedModel
7
+ model = tf.keras.models.load_model("waste_sort_model.keras")
8
+
9
  class_names = ["Non-Recyclable", "Recyclable"]
10
 
11
  def classify_image(img):