Update app.py

app.py

@@ -4,7 +4,7 @@ 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):
+def dynamic_label(ground_truth_class):
     return [gr.Image(label=f"Generated Image of {ground_truth_class}") for _ in range(NUM_GEN)]
 
 
@@ -16,7 +16,7 @@ iface = gr.Interface(
         gr.Textbox(label="Mistaken Class"),
         gr.Textbox(label="Ground Truth Class")
     ],
-    outputs=dynamic_label,
+    outputs=dynamic_label(inputs[3]),
     title="visual-data-aug",
     live=True,
 )