ncncomplete commited on
Commit
54a608e
·
verified ·
1 Parent(s): 0d442f1

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import tensorflow as tf
4
  import numpy as np
5
 
6
  model = tf.keras.models.load_model('model.h5')
7
- labels = ['flower_photos'] # Update this after training
8
 
9
  def predict(image):
10
  image = tf.image.resize(image, (224, 224))
 
4
  import numpy as np
5
 
6
  model = tf.keras.models.load_model('model.h5')
7
+ labels = ['Class_A', 'Class_B'] # Update this after training
8
 
9
  def predict(image):
10
  image = tf.image.resize(image, (224, 224))