Update app.py

app.py

@@ -27,7 +27,7 @@ def infer(prompt):
           )
     images_edited = out_edit.images[0]
     
-    return [images, images_edited]
+    return [(images, 'Stable Diffusion'), (images_edited, 'Fair Diffusion')]
     
 
 css = """