add model details

.gitignore

@@ -0,0 +1,5 @@
+.idea
+.DS_Store
+__pycache__
+**logging**
+**__pycache__**

app.py

@@ -55,14 +55,29 @@ def generate(prompt: str,
         return [output,]
 
 
+candidates = [
+    "0. [JuggernautReborn] realistic",
+    "1. [MajicmixRealisticV7] realistic, Asia portrait",
+    "2. [EpicRealismV5] realistic",
+    "3. [RealisticVisionV5] realistic",
+    "4. [MajicmixFantasyV3] animation",
+    "5. [MinimalismV2] illustration",
+    "6. [RealCartoon3dV17] cartoon 3d",
+    "7. [AWPaintingV1.4] animation",
+]
 
 def main():
     with gr.Blocks() as demo:
-        gr.Markdown("# DMM")
+        gr.Markdown(
+            """
+            # DMM Demo
+            The checkpoint is https://huggingface.co/MCG-NJU/DMM.
+            """
+        )
         with gr.Row():
             with gr.Column():
                 with gr.Column():
-                    model_id = gr.Dropdown(list(range(8)), label="Model Index")
+                    model_id = gr.Dropdown(candidates, label="Model Index", type="index")
                     all_check = gr.Checkbox(label="All (ignore the selection above)")
                 prompt = gr.Textbox("portrait photo of a girl, long golden hair, flowers, best quality", label="Prompt")
                 negative_prompt = gr.Textbox("worst quality,low quality,normal quality,lowres,watermark,nsfw", label="Negative Prompt")
@@ -70,7 +85,7 @@ def main():
                     seed = gr.Number(0, label="Seed", precision=0, scale=3)
                     update_seed_btn = gr.Button("🎲", scale=1)
                 with gr.Row():
-                    height = gr.Number(512, step=8, label="Height (suggest 512~768)")
+                    height = gr.Number(768, step=8, label="Height (suggest 512~768)")
                     width = gr.Number(512, step=8, label="Width")
                 submit_btn = gr.Button("Submit", variant="primary")
             output = gr.Gallery(label="images")
@@ -81,7 +96,7 @@ def main():
         update_seed_btn.click(lambda: random.randint(0, 1000000),
                               outputs=[seed])
 
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch()
 
 
 if __name__ == "__main__":