Hugging Face's logo

Spaces:
Sa-m
/
Smart-Pet-Classifier
Sleeping

App Files Community
Sa-m commited on
Commit
3b6db67
•
1 Parent(s): ec863f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ model=load_model('Models/best_model1.h5')
43
  def classify_image(inp):
44
  NUM_CLASSES=2
45
  # Resize the image to the required size
46
- labels = {'Cat': 0, 'Dog': 1}
47
  inp = tf.image.resize(inp, [IMG_SIZE, IMG_SIZE])
48
  inp = inp.numpy()
49
  inp = inp.reshape((-1, IMG_SIZE, IMG_SIZE, 3))
 
43
  def classify_image(inp):
44
  NUM_CLASSES=2
45
  # Resize the image to the required size
46
+ labels = ['Cat','Dog']
47
  inp = tf.image.resize(inp, [IMG_SIZE, IMG_SIZE])
48
  inp = inp.numpy()
49
  inp = inp.reshape((-1, IMG_SIZE, IMG_SIZE, 3))