Update app.py

app.py

@@ -129,7 +129,7 @@ def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATI
     
     print(classifier)
     
-    return image_paths, classifier[0]
+    return image_paths, {labels[i]: classifier[0][i] for i in range(len(labels))}
 
 def cloud():
     print("[CLOUD] | Space maintained.")