Hugging Face's logo

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

App Files Community
Sa-m commited on
Commit
57ac805
·
verified ·
1 Parent(s): b6ccd3d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -29,8 +29,8 @@ def classify_image(inp):
29
  np.random.seed(143)
30
  labels = {'Cat': 0, 'Dog': 1}
31
  NUM_CLASSES = 2
32
- inp = inp.reshape((-1, HEIGHT, WIDTH, 3))
33
- inp = tf.keras.applications.nasnet.preprocess_input(inp)
34
  prediction = model.predict(inp)
35
  predicted_class_indices = np.argmax(prediction, axis=1)
36
 
 
29
  np.random.seed(143)
30
  labels = {'Cat': 0, 'Dog': 1}
31
  NUM_CLASSES = 2
32
+ #inp = inp.reshape((-1, HEIGHT, WIDTH, 3))
33
+ #inp = tf.keras.applications.nasnet.preprocess_input(inp)
34
  prediction = model.predict(inp)
35
  predicted_class_indices = np.argmax(prediction, axis=1)
36