Update app.py

app.py

@@ -86,7 +86,7 @@ def generate(
         output_type="battery",
     ).images
 
-    return output
+    return output, [[prompt, seed, output[0]]]
 
 
 @spaces.GPU
@@ -126,7 +126,7 @@ def img2img_generate(
         output_type="battery",
     ).images
 
-    return output
+    return output, [[prompt, seed, output[0]]]
 
 
 examples = [
@@ -264,7 +264,7 @@ with gr.Blocks(css=css) as demo:
                     steps,
                     number_image,
                 ],
-                outputs=[result],
+                outputs=[result, generated_table],
                 api_name="run",
             )
         
@@ -296,7 +296,7 @@ with gr.Blocks(css=css) as demo:
                     strength,
                     number_image,
                 ],
-                outputs=[img2img_output],
+                outputs=[img2img_output, generated_table],
                 api_name="img2img_run",
             )
     
@@ -308,13 +308,11 @@ with gr.Blocks(css=css) as demo:
             </p>
             """
         )
-        generated_table = gr.Table(
+        generated_table = gr.Dataframe(
             headers=["Prompt", "Seed", "Image"],
-            row_count=1,
+            datatype=["str", "number", "image"],
             col_count=(3, "fixed"),
-            type="gallery",
-            show_label=False,
-            elem_id="generated-table",
+            row_count=1,
         )
 
 if __name__ == "__main__":