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