zxcgqq commited on
Commit
9ceef03
·
1 Parent(s): 5b9233e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "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']
 
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']