Update app.py

app.py

@@ -11,6 +11,7 @@ import tensorflow_hub as hub
 
 
 IMAGE_DIM = 299   # required/default image dimensionality
+model = load_model("nsfw.299x299.h5")  
 
 def load_images(image_paths, image_size, verbose=True):
     loaded_images = []
@@ -58,9 +59,9 @@ def classify_nd(model, nd_images, predict_args={}):
 
 
 def nsfw(image):
-    model = load_model("nsfw.299x299.h5")    
-    # image_preds = classify_nd(model, image)
-    return "----"
+     
+    image_preds = classify_nd(model, image)
+    return image_preds
 
 
 demo = gr.Interface(fn=nsfw,