Update app.py

app.py

@@ -80,8 +80,8 @@ def train_image_generation_model(image_folder, text_folder, model_name="image_ge
 iface = gr.Interface(
     fn=train_image_generation_model,
     inputs=[
-        gr.File(label="Image Folder"),
-        gr.File(label="Text Prompts Folder"),
+        gr.File(label="Image Folder", file_count="directory"),
+        gr.File(label="Text Prompts Folder", file_count="directory"),
         gr.Textbox(label="Model Name"),
     ],
     outputs=gr.File(label="Model File"),