Update app.py

app.py

@@ -3,6 +3,11 @@ import gradio as gr
 from src.v2_for_hf import generate_images
 from src.v2_for_hf import NUM_GEN
 
+
+def dynamic_label(api_key, input_image, mistaken_class, ground_truth_class):
+    return [gr.Image(label=f"Generated Image of {ground_truth_class}") for _ in range(NUM_GEN)]
+
+
 iface = gr.Interface(
     fn=generate_images,
     inputs=[
@@ -11,10 +16,9 @@ iface = gr.Interface(
         gr.Textbox(label="Mistaken Class"),
         gr.Textbox(label="Ground Truth Class")
     ],
-    outputs=[
-        gr.Image(label="Output Image") for i in range(NUM_GEN)
-    ],
+    outputs=dynamic_label,
     title="visual-data-aug",
+    live=True,
 )
 
 if __name__ == "__main__":