Update app.py
Browse files
app.py
CHANGED
@@ -52,7 +52,7 @@ def load_images(image_paths, image_size, verbose=True):
|
|
52 |
|
53 |
def classify_nd(model, nd_images, predict_args={}):
|
54 |
file_path = "test.png"
|
55 |
-
img =
|
56 |
model_preds = model.predict(img, **predict_args)
|
57 |
categories = ['drawings', 'hentai', 'neutral', 'porn', 'sexy']
|
58 |
|
|
|
52 |
|
53 |
def classify_nd(model, nd_images, predict_args={}):
|
54 |
file_path = "test.png"
|
55 |
+
img = load_images(file_path,(299,299))
|
56 |
model_preds = model.predict(img, **predict_args)
|
57 |
categories = ['drawings', 'hentai', 'neutral', 'porn', 'sexy']
|
58 |
|