Kvikontent commited on
Commit
e88f579
1 Parent(s): b33888d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -82,10 +82,11 @@ iface = gr.Interface(
82
  inputs=[
83
  gr.File(label="Image Folder"),
84
  gr.File(label="Text Prompts Folder"),
 
85
  ],
86
  outputs=gr.File(label="Model File"),
87
  title="Image Generation Model Trainer",
88
- description="Upload a folder of images and their corresponding text prompts to train a model.",
89
  )
90
 
91
  iface.launch(share=True)
 
82
  inputs=[
83
  gr.File(label="Image Folder"),
84
  gr.File(label="Text Prompts Folder"),
85
+ gr.Textbox(label="Model Name"),
86
  ],
87
  outputs=gr.File(label="Model File"),
88
  title="Image Generation Model Trainer",
89
+ description="Upload a folder of images and their corresponding text prompts to train a model.\n Images foler should contain image files. Prompts folder should contain .txt files. Each text file is prompt for each image in images folder.",
90
  )
91
 
92
  iface.launch(share=True)