Update app.py

app.py

@@ -131,11 +131,7 @@ def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATI
 
     print(nsfw_prediction)
 
-    nsfw_format = [f"{index['label']}: {index['score']:.3%}" for index in nsfw_prediction]
-    
-    print(nsfw_format)
-    
-    return image_paths, nsfw_format
+    return image_paths, {item['label']: round(item['score'], 3) for item in nsfw_prediction}
 
 def cloud():
     print("[CLOUD] | Space maintained.")