Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ def normalize_image(img):
|
|
16 |
|
17 |
def predict_top_classes(img, num_top_classes=5):
|
18 |
img = img.convert('RGB')
|
19 |
-
img_data =
|
20 |
x = np.array(img_data)
|
21 |
x = np.expand_dims(x, axis=0)
|
22 |
temp = model.predict(x)
|
|
|
16 |
|
17 |
def predict_top_classes(img, num_top_classes=5):
|
18 |
img = img.convert('RGB')
|
19 |
+
img_data = normalize_image(img)
|
20 |
x = np.array(img_data)
|
21 |
x = np.expand_dims(x, axis=0)
|
22 |
temp = model.predict(x)
|