Update app.py

app.py

@@ -3,11 +3,6 @@ import gradio as gr
 from src.v2_for_hf import generate_images
 from src.v2_for_hf import NUM_GEN
 
-
-def dynamic_label(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=[
@@ -16,9 +11,10 @@ iface = gr.Interface(
         gr.Textbox(label="Mistaken Class"),
         gr.Textbox(label="Ground Truth Class")
     ],
-    outputs=dynamic_label(inputs[3]),
+    outputs=[
+        gr.Image(label="Generated Image of Ground Truth Class") for i in range(NUM_GEN)
+    ],
     title="visual-data-aug",
-    live=True,
 )
 
 if __name__ == "__main__":