Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -41,6 +41,7 @@ model=load_model('Models/best_model1.h5')
|
|
| 41 |
# return result
|
| 42 |
|
| 43 |
def classify_image(inp):
|
|
|
|
| 44 |
# Resize the image to the required size
|
| 45 |
inp = tf.image.resize(inp, [IMG_SIZE, IMG_SIZE])
|
| 46 |
inp = inp.numpy()
|
|
|
|
| 41 |
# return result
|
| 42 |
|
| 43 |
def classify_image(inp):
|
| 44 |
+
NUM_CLASSES=2
|
| 45 |
# Resize the image to the required size
|
| 46 |
inp = tf.image.resize(inp, [IMG_SIZE, IMG_SIZE])
|
| 47 |
inp = inp.numpy()
|