Update app.py

app.py

@@ -123,12 +123,13 @@ def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATI
     images = repo(**parameters).images
     image_paths = [save_image(img, seed) for img in images]
 
+    print(image_paths)
+
     classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection")(Image.open(image_paths[0]))
     
-    print(image_paths)
     print(classifier)
     
-    return image_paths, classifier
+    return image_paths, classifier[0]
 
 def cloud():
     print("[CLOUD] | Space maintained.")