Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -60,7 +60,7 @@ def classify_nd(model, nd_images, predict_args={}):
|
|
60 |
def nsfw(image):
|
61 |
model = load_model("nsfw.299x299.h5")
|
62 |
image_preds = classify_nd(model, image)
|
63 |
-
return
|
64 |
|
65 |
|
66 |
demo = gr.Interface(fn=nsfw,
|
|
|
60 |
def nsfw(image):
|
61 |
model = load_model("nsfw.299x299.h5")
|
62 |
image_preds = classify_nd(model, image)
|
63 |
+
return image_preds
|
64 |
|
65 |
|
66 |
demo = gr.Interface(fn=nsfw,
|